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. 最佳买卖股票时机含冷冻期 动态规划代码

上一篇:【LeetCode刷题(中等程度)】309. 最佳买卖股票时机含冷冻期


下一篇:Assertion failed: new_time >= loop->time, file c:\ws\deps\uv\src\win\core.c, line 309