LeetCode309最佳买卖股票时机含冷冻期(动态规划)

题目链接:leetcode309

LeetCode309最佳买卖股票时机含冷冻期(动态规划)_第1张图片

思路:因为本题中规定买入,卖出和冷冻操作,规定上一次卖出和下一次买入之间需要至少一天的冷冻期,因此我们定义三种状态sell、buy和cooldown,分别对应了到第i天为止最后一个操作是买入、卖出和冷冻所对应的最大利润,则状态转移方程如下:

  1. sell[i]=max(sell[i-1],buy[i-1]+prices[i]);
  2. buy[i]=max(buy[i-1],cooldown[i-1]-prices[i]);
  3. cooldown[i]=max(cooldown[i-1],max(sell[i-1],buy[i-1])).

Code

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if(prices.empty()) return 0;
        int n=prices.size(); //数量
        vector<int> sell(n,0); //手中不持有股票
        vector<int> buy(n,0); //手中持有股票
        vector<int> cooldown(n,0); //此时为冷冻期(可能持有也可能不持有)的最大利润
        buy[0]=-prices[0]; //根据定义第一天持有股票就一定要购买
        for(int i=1;i<n;i++){
            sell[i]=max(sell[i-1],buy[i-1]+prices[i]); //不持有股票要
            buy[i]=max(buy[i-1],cooldown[i-1]-prices[i]); //持有股票由于有冷冻期的原因要经过一个冷冻期才能够再次购买
            cooldown[i]=max(cooldown[i-1],max(sell[i-1],buy[i-1])); //更新冷冻期的最大利润
        }
        return sell.back();
    }
};

你可能感兴趣的:(DP,LeetCode)