309. 最佳买卖股票时机含冷冻期(二维dp)

Posted 2020-12-09 vampire6

309. 最佳买卖股票时机含冷冻期

 

 解释见代码：

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if(prices.size()<=1)
            return 0;
        int dp[prices.size()+10][3];//第一维：天数  第二维：0：手上没票 1：手上有票 2：进入冷冻期（区间冷冻期第一天）
        dp[0][0]=0;//第一天手上没票 
        dp[0][1]=-prices[0];//第一天手上有票负成本利润
        dp[0][2]=0;//第一天进入冷冻
        for(int i=1;i<prices.size();i++)
        {
            dp[i][0]=max(dp[i-1][0],dp[i-1][2]);
            //当前手上没票：1.前一天手上也没票  2.前一天处在冷冻期（非冷冻期的第一天）
            dp[i][1]=max(dp[i-1][1],dp[i-1][0]-prices[i]);
            //当天手上有票：1.前一天手上有票    2.前一天手上没票，今天买入
            dp[i][2]=dp[i-1][1]+prices[i];
            //当天为这次冷冻期第一天即前一天有票今天卖掉
        }
        return max(dp[prices.size()-1][0],dp[prices.size()-1][2]);//返回最后一天手上没票和是处于冷冻期的最大利润
    }
};