Leetcode#123 Best Time to Buy and Sell Stock III

原题地址

 

最直观的想法就是划分成两个子问题,每个子问题变成了:求在某个范围内交易一次的最大利润

在只能交易一次的情况下,如何求一段时间内的最大利润?其实就是找股价最低的一天买进,然后在股价最高的一天卖出,当然,卖股票要在买股票之后。

用迭代法求这个问题非常容易,令profits[i]表示截至到第i天的最大利润,则profits[i+1] = max{profits[i], prices[i+1] - minPrice},其中,minPrice = min{prices[k]},0 <= k < n。但是从profits[i]没法直接得到profits[i-1]

倒过来迭代也完全没问题,令profits[i]表示从第i天开始到结束的最大利润,则profits[i] = max{profits[i+1], maxPrice - prices[i]},其中maxPrice = max{prices[k]},i+1 <= k < n。但是从profits[i]没法直接得到profits[i+1]

 

之后,依次枚举分割点,求解两个子问题的和即可。

 

代码:

 1 int maxProfit(vector<int> &prices) {
 2   int n = prices.size();
 3   vector<int> profits(prices.size(), 0);
 4   int result = 0;
 5   int tmp;
 6         
 7   if (n < 2)
 8     return 0;
 9 
10   tmp = 0;
11   int minPrice = prices[0];
12   for (int i = 1; i < n; i++) {
13     tmp = max(tmp, prices[i] - minPrice);
14     profits[i] = tmp;
15     minPrice = min(minPrice, prices[i]);
16   }
17         
18   tmp = 0;
19   int maxPrice = prices[n - 1];
20   for (int i = n - 2; i >= 0; i--) {
21     tmp = max(tmp, maxPrice - prices[i]);
22     result = max(result, i > 0 ? tmp + profits[i - 1] : tmp);
23     maxPrice = max(maxPrice, prices[i]);
24   }
25         
26   return result;
27 }

 

posted @ 2015-01-21 13:08  李舜阳  阅读(882)  评论(0编辑  收藏  举报