买卖股票的最佳时机(含冷冻期)

买卖股票的系列问题五！ 这回的限制是含有冷冻期。

导读 ^ _ ^

买卖股票的系列问题五！
这回的限制是含有冷冻期。

题目

leetcode 309

代码与思路

确定状态

明确状态关系，四种
冷冻期，卖出股票后不能再买一次
dp[i][j]，第i天状态为j，所剩的最多现金为dp[i][j]。

状态0：买入股票状态（今天买入股票，或者是之前就买入了股票然后没有操作） 卖出股票状态，这里就有两种卖出股票状态 状态1：两天前就卖出了股票，度过了冷冻期，一直没操作，今天保持卖出股票状态 状态2：今天卖出了股票 状态3：今天为冷冻期状态，但冷冻期状态不可持续，只有一天！

 dp[i][0] = max(dp[i - 1][0], max(dp[i - 1][3], dp[i - 1][1]) - prices[i]);
//前一天持有，或者前一天是冷冻期或者前一天保持卖出状态
dp[i][1] = max(dp[i - 1][1], dp[i - 1][3]);
//前一天是卖出状态或者前一天是冷冻期
dp[i][2] = dp[i - 1][0] + prices[i];
//卖
dp[i][3] = dp[i - 1][2];
//前一天是卖出状态
 

初始化

如果是持有股票状态（状态0）那么：dp[0][0] = -prices[0]，买入股票所剩现金为负数。
保持卖出股票状态（状态1），第0天没有卖出dp[0][1]初始化为0就行，
今天卖出了股票（状态2），同样dp[0][2]初始化为0，因为最少收益就是0，绝不会是负数。
同理dp[0][3]也初始为0。

遍历答案

 //买卖股票的最佳时机（含冷冻期）

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        if (n == 0) return 0;
        vector<vector<int>> dp(n, vector<int>(4, 0));
        dp[0][0] -= prices[0]; // 持股票
        for (int i = 1; i < n; i++) {
            dp[i][0] = max(dp[i - 1][0], max(dp[i - 1][3], dp[i - 1][1]) - prices[i]);
            dp[i][1] = max(dp[i - 1][1], dp[i - 1][3]);
            dp[i][2] = dp[i - 1][0] + prices[i];
            dp[i][3] = dp[i - 1][2];
        }
        //返回几个状态中的最大值
        return max(dp[n - 1][3],max(dp[n - 1][1], dp[n - 1][2]));
    }
};
 

#谢谢你的观看！

^ _ ^

查看更多关于买卖股票的最佳时机(含冷冻期)的详细内容...