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

https://leetcode-cn.com/problems/best-time-to-buy-and-sell-stock-with-cooldown/

枚举型

dp[i][j]，前i个，j：0买入，1卖出，冷冻不用管，由于是枚举上一次操作的日期，买入的上一次卖出是前2天之前。

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n=prices.size();
        if(n==0)
            return 0;
        int dp[n][2];
        memset(dp,0,sizeof(dp));
        for(int i=0;i<n;i++){
            dp[i][0]=0;
            for(int j=0;j<i;j++){
                if(j<i){
                    dp[i][1]=max(dp[i][1],dp[j][0]+prices[i]-prices[j]);
                }
                
                if(j<i-1){
                    dp[i][0]=max(dp[i][0],dp[j][1]);
                }
            }
        }
        int ans=0;
        for(int i=0;i<n;i++){
            ans=max(ans,dp[i][1]);
        }
        return ans;
    }
};

 

状态型

dp[i][j]，前i个，j：0不持股，1持股，2冷冻期。复杂度O（n），买入的时候减一个数，卖出的时候加一个数，合起来就是一个差。

然后状态是完成动作后的状态，这样是因为对于未来的读取过去转移更方便。

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n=prices.size();
        if(n==0)
            return 0;
        int dp[n][3];
        memset(dp,0,sizeof(dp));
        
        dp[0][1]=-prices[0];

        for(int i=1;i<n;i++){
            dp[i][0]=max(dp[i-1][0],dp[i-1][1]+prices[i]);
            dp[i][1]=max(dp[i-1][1],dp[i-1][2]-prices[i]);
            dp[i][2]=dp[i-1][0];
        }

        return max(dp[n-1][0],dp[n-1][2]);
    }
};