Chinaunix首页 | 论坛 | 博客

分类: C/C++

2018-03-18 21:34:03

这阵子一直在刷题,刷到这个系列了,写点自己能理解的心得体会吧,不是最优的,但个人认为理解的不太难。
言归征战,这个系列的大体思路就是给你一连串的股票价格,再加上各个限定条件,让你求出每种限制条件下,各自能获得最大的利润。(要是能有这个预知未来的能力,可以不用搬砖了,orz...)
I. 限制条件:只允许交易一次;
思路:先算出每次交易的利润,正表示盈余,负表示亏损,问题就转换为累加数组,求连续子数组的最大值,代码如下:

点击(此处)折叠或打开

  1. int maxProfiti(vector<int>& prices){
  2.     int len = prices.size(), res = 0, i;
  3.     if(len < 2) return 0;
  4.     vector<int> profit(len, 0);
  5.     for(i = 0; i < len-1; i++)
  6.         profit[i] = prices[i+1]-prices[i];
  7.     for(i=len-2; i >=0; i--){
  8.         if(profit[i+1]>0)
  9.             profit[i] += profit[i+1];
  10.         res = res > profit[i] ? res: profit[i];
  11.     }
  12.     return res;
  13. }
II.限制条件:允许进行尽可能多的交易;
思路:同上,求得利润数组,然后遍历,正数就加,负数就过,代码如下:

点击(此处)折叠或打开

  1. int maxProfitii(vector<int>& prices){
  2.     int len = prices.size(), res = 0, i;
  3.     if(len < 2) return 0;
  4.     vector<int> profit(len, 0);
  5.     for(i = 0; i < len-1; i++)
  6.         profit[i] = prices[i+1]-prices[i];
  7.     for( i = 0; i < len; i++){
  8.         if(profit[i] > 0)
  9.             res += profit[i];
  10.     }
  11.     return res;
  12. }
III. 限制条件,最多允许交易两次;
思路:本来见到这题,想用上边题I的解题思路,dp[i][j]表示从i到j(包含)只允许一次交易的最大利润,但是时间复杂度太高,看了后面讨论区,恍然大悟,dp[k][i]表示到价格i为止,允许最多k次交易时的最大利润,则递推公式为dp[k][i]=max(dp[k][i-1], prices[i]-prices[j]+dp[k-1][j]),j的范围为[0, i],代码如下:

点击(此处)折叠或打开

  1. int maxProfitiii(vector<int>& prices){
  2.     int len = prices.size(), K = 2, temp, k, i;
  3.     if(len < 2) return 0;
  4.     vector<vector<int>> dp(K+1, vector<int>(len, 0));
  5.     for(k = 1; k <= K; k++){
  6.         temp = dp[k-1][0] - prices[0];
  7.         for(i = 1; i < len; i++){
  8.             dp[k][i] = max(dp[k][i-1], prices[i] +temp);
  9.             temp = max(temp, dp[k-1][i]-prices[i]);
  10.         }
  11.     }
  12.     return dp[K][len-1];
  13. }


在代码里面耍了一个trick,除去了遍历dp[k-1][0]到dp[k-1][i]这一增加时间的代码块:);
IV.限制条件:最多进行k次交易;
思路:有了三题的经验,第四题就手到擒来了,但是要注意k与所给数组长度的比较,因为如果2*k>=len 的话,就转换为题目II了,代码如下:

点击(此处)折叠或打开

  1. int maxProfitiv(int k, vector<int>& prices){
  2.     int len = prices.size(), K , temp, i, res= 0;
  3.     if(len < 2) return 0;
  4.     if(len<=2*k){
  5.             for(i = 1; i < len; i++)
  6.                 if(prices[i]>prices[i-1])
  7.                     res +=(prices[i]-prices[i-1]);
  8.             return res;
  9.         }
  10.     vector<int> dp(len, 0);
  11.     for(K = 1; K <= k; K++){
  12.         temp = dp[0] - prices[0];
  13.         for(i = 1; i < len; i++){
  14.             temp = max(temp, dp[i]-prices[i]);
  15.             dp[i] = max(dp[i-1], prices[i] +temp);
  16.         }
  17.     }
  18.     return dp[len-1];
  19. }
V.限制条件:每次交易有交易费用fee,允许尽可能多的交易;
思路:刚开始以为顺着第四题思路,将第15行,改为dp[i] = max(dp[i-1], prices[i] +temp-fee);即可,但是超时了...继续讨论区,看大神--->建两个数组hold[i], sold[i],表示到第i天,手中握有股票的最大利润和将第i天的股票卖出的最大利润,递推如下:
hold[i] = max(hold[i-1], sold[i-1]-prices[i]);
sold[i] = max(sold[i-1], hold[i-1]+prices[i]-fee);
代码如下:

点击(此处)折叠或打开

  1. int maxProfitiv( vector<int>& prices, int fee){
  2.     int len = prices.size(), i;
  3.     vector<int> hold(len, 0), sold(len, 0);
  4.     hold[0] = -prices[0];
  5.     for(i = 1; i < len; i++){
  6.         hold[i] = max(hold[i-1], sold[i-1]-prices[i]);
  7.         sold[i] = max(sold[i-1], hold[i-1]+prices[i]-fee);
  8.     }
  9.     return sold[len-1];
  10. }
VI.限制条件:每次卖出后,必须冷静一天;
思路:哎,不知道为啥还得休整一天,顺着V题思路(没错,又是顺着思路。。。),hold[i]只与sold[i-2]有关系,代码修改如下:

点击(此处)折叠或打开

  1. int maxProfitvi(vector<int>& prices){
  2.     int len = prices.size(), i;
  3.     if(len < 2) return 0;
  4.     vector<int> hold(len, 0), sold(len, 0);
  5.     hold[0] = -prices[0];
  6.     hold[1] = max(hold[0], -prices[1]);
  7.     sold[1] = max(sold[0], hold[1]+prices[1]);
  8.     for(i = 2; i < len; i++){
  9.         hold[i] = max(hold[i-1], sold[i-2]-prices[i]);
  10.         sold[i] = max(sold[i-1], hold[i-1]+prices[i]);
  11.     }
  12.     return sold[len-1];
  13. }
总结:有些题目条件变化不大,可以顺着思路继续走,但是一旦问题发生的变化使得你之前的解题思路的总体假设不再成立,必须考虑换思路了。。。哎,路漫漫其修远兮,吾将上下而求索~~~继续努力!!!
阅读(320) | 评论(0) | 转发(0) |
给主人留下些什么吧!~~
评论热议
请登录后评论。

登录 注册