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

[LeetCode] Best Time to Buy and Sell Stock III

时间:2014-07-06 23:40:44      阅读:284      评论:0      收藏:0      [点我收藏+]

标签:style   blog   color   问题   for   io   

 将Best Time to Buy and Sell Stock的如下思路用到此题目

思路1:第i天买入,能赚到的最大利润是多少呢?就是i + 1 ~ n天中最大的股价减去第i天的。

思路2:第i天买出,能赚到的最大利润是多少呢?就是第i天的价格减去 0~ i-1天中最小的。

 

和前两道题比起来的话,这道题最难了,因为限制了交易次数。
解决问题的途径我想出来的是:既然最多只能完成两笔交易,而且交易之间没有重叠,那么就divide and conquer。
设i从0到n-1,那么针对每一个i,看看在prices的子序列[0,...,i][i,...,n-1]上分别取得的最大利润(第一题)即可。
这样初步一算,时间复杂度是O(n2)。


改进:
改进的方法就是动态规划了,那就是第一步扫描,先计算出子序列[0,...,i]中的最大利润,用一个数组保存下来,那么时间是O(n)。
第二步是逆向扫描,计算子序列[i,...,n-1]上的最大利润,这一步同时就能结合上一步的结果计算最终的最大利润了,这一步也是O(n)。
所以最后算法的复杂度就是O(n)的。

 

/*
解释:
首先,因为能买2次(第一次的卖可以和第二次的买在同一时间),但第二次的买不能在第一次的卖左边。
所以维护2个表,f1和f2,size都和prices一样大。
意义:
f1[i]表示 -- 截止到i下标为止,左边所做交易能够达到最大profit;[0,...,i]的利润
f2[i]表示 -- 截止到i下标为止,右边所做交易能够达到最大profit;[i,...,n-1]的利润
那么,对于f1 + f2,寻求最大即可。
*/

 对于f1[i],求解过程中用price[i] 减去之前的最小值 和 f1[i-1]做比较,取最大值

动态规划转移方程 f1[i] = max(f1[i-1], price[i]- min)

对于f2[i],求解过程中用后面的最大值减去price[i]和f2[i+1]做比较,取最大值

 动态规划转移方程 f2[i] = max(f2[i+1], max-price[i])

 

思路解释完毕,上code:

minX[i] 表示0 到 i 的最小值 的price

max[i] 表示i到n-1的最大值的price

 1 class Solution {
 2     public:
 3         int maxProfit(vector<int> &prices) {
 4             if (prices.size() == 0)
 5                 return 0;
 6 
 7             vector<int> f1(prices.size());
 8             vector<int> f2(prices.size());
 9 
10             vector<int> minX(prices.size());
11             vector<int> maxX(prices.size());
12 
13             minX[0] = prices[0];
14             for(int i = 1; i< prices.size();i++ )
15             {
16                 minX[i] = min(minX[i-1], prices[i]);
17             }
18 
19             maxX[prices.size()-1] = prices[prices.size()-1];
20             for(int i = prices.size() -2; i >=0; i-- )
21             {
22                 maxX[i] = max(maxX[i+1], prices[i]);
23             }
24 
25             f1[0] = 0;
26             for(int i = 1; i< prices.size();i++ )
27             {
28                 f1[i] = max(f1[i-1],prices[i]-minX[i]);
29             }
30 
31             f2[prices.size()-1] = 0;
32             for(int i = prices.size() -2; i >=0; i-- )
33             {
34                 f2[i] = max(f2[i+1],maxX[i]- prices[i]);
35             }
36 
37 
38             int sum = 0;
39 
40             for(int i = 1; i< prices.size();i++ )
41                 sum = max(sum, f1[i] + f2[i]);
42 
43             return sum;
44 
45         }
46 };

 

 

优化:可以对上述code稍微优化一下,maxX 和minX array 并不需要,只要保留一个变量mini和maxX即可,不过由于f1和f2 的存在,空间复杂度还是O(n)

 

 

 1 class Solution {
 2     public:
 3         int maxProfit(vector<int> &prices) {
 4             if (prices.size() == 0)
 5                 return 0;
 6 
 7             vector<int> f1(prices.size());
 8             vector<int> f2(prices.size());
 9     
10             int mini = prices[0];
11             f1[0] = 0;
12             for(int i = 1; i< prices.size();i++ )
13             {   
14                 f1[i] = max(f1[i-1],prices[i]-mini);
15                 mini = min(mini, prices[i]);
16             }   
17 
18             int maxi = prices[prices.size()-1];
19             f2[prices.size()-1] = 0;
20             for(int i = prices.size() -2; i >=0; i-- )
21             {   
22                 f2[i] = max(f2[i+1],maxi - prices[i]);
23                 maxi = max(maxi, prices[i]);
24             }   
25 
26             int sum = 0;
27 
28             for(int i = 1; i< prices.size();i++ )
29                 sum = max(sum, f1[i] + f2[i]);
30 
31             return sum;
32 
33         }   
34 };

 

[LeetCode] Best Time to Buy and Sell Stock III,布布扣,bubuko.com

[LeetCode] Best Time to Buy and Sell Stock III

标签:style   blog   color   问题   for   io   

原文地址:http://www.cnblogs.com/diegodu/p/3822539.html

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