Say you have an array for which the ith element
is the price of a given stock on day i.
Design an algorithm to find the maximum profit. You may complete at most two transactions.
Note:
You may not engage in multiple transactions at the same time (ie, you must sell the stock before you buy again).
现在条件变成了最多两次,一个直接的思路是枚举切分点,然后分别计算前后只能买一次的最大值,相加;当然也可能整个只买一次。
直观看复杂度是o(n2)的,试了下大数据果然是过不了~
其实遇到这种最多两次的题,要敏感,最多两次跟最多K次还不一样,最多两次的题通常都可以转化为一次的情况,即前向扫描一遍,保留profit[i],再后向扫描一遍,扫描到i的时候,就是这次扫描的最大值加前向扫描的那个profit[i]之和。
代码:
class Solution { public: int maxProfit(vector<int>& price) { int n=price.size(); if (n<=1) return 0; vector<int> profit(n,0); int low=price[0],i; for(i=1;i<n;i++) { if (price[i]>low) { profit[i]=max(profit[i-1],price[i]-low); } else { profit[i]=profit[i-1]; low=price[i]; } } int high=price[n-1]; int ans=0; for(i=n-1;i>=0;i--) { if (high>price[i]) { ans=max(ans,profit[i]+high-price[i]); } else high=price[i]; } return ans; } };
人家一样的思路为什么16ms过大合集,我的要60ms啊!!!!!
是因为使用了max函数吗? 哪位同学知道