309. 最佳买卖股票时机含冷冻期 动态规划代码

题的传送门：309. 最佳买卖股票时机含冷冻期

dp 动态规划
have[i] 表示第 i 天持有股票，那么就只能是今天买入或者之前就有
no[i] 表示第 i 天没有持有股票，那么就只能是今天卖出或者之前就没有
当前题为无限次购买，不过存在冷冻期而已
状态转移方程：
1、普通的无限次股票购买：
have[i] = Math.max(have[i - 1], no[i - 1] - prices[i]);
no[i] = Math.max(no[i - 1], have[i - 1] + prices[i]);
2、含冷冻期的无限次股票购买
have[i] = Math.max(have[i - 1], no[i - 2] - prices[i]); 
no[i] = Math.max(no[i - 1], have[i - 1] + prices[i]);
差别在于 have[i] 方程中的 no[i - 1] 和 no[i - 2]
no[i - 1] 表示昨天没有持有股票，昨天没有持有的话有两种可能：
1、之前都没有持有，那么 no[i - 1] = no[i - 2]
2、昨天卖出股票，那么今天就是冷冻期了，那么今天不能购买
因此如果今天要购买的话，那么只能是选择昨天没有卖出股票的情况，即 no[i - 2]
初始条件：
have[0] = -prices[0], no[0] = 0
have[0] 表示如果第一天持有的话只能买入，收益为 -prices[0]
no[0] 表示第一天不持有，那么无作为，收益为 0

代码如下：

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        if(n == 0)
            return 0;
        
        vector<int> have(n);
        vector<int> no(n);

        have[0] = -prices[0];
        no[0] = 0;
        for(int i = 1; i < n; i++){
            have[i] = max(have[i-1], i>=2? no[i-2] - prices[i]: -prices[i]);
            no[i] = max(no[i-1],have[i - 1] + prices[i]);
        }
        return no[n-1];
    }
};

标签：309,no,max,代码,股票,prices,冷冻,Math
来源： https://www.cnblogs.com/cherryg/p/14651377.html