码迷,mamicode.com
首页 > 其他好文 > 详细

Leetcode#123 Best Time to Buy and Sell Stock III

时间:2015-01-21 15:01:30      阅读:165      评论:0      收藏:0      [点我收藏+]

标签:

原题地址

 

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

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

用迭代法求这个问题非常容易,令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 }

 

Leetcode#123 Best Time to Buy and Sell Stock III

标签:

原文地址:http://www.cnblogs.com/boring09/p/4238606.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!