309. [C++]最佳买卖股票时机含冷冻期

题目描述

给定一个整数数组，其中第 i 个元素代表了第 i 天的股票价格 。​

设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票）:

你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。
卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。
示例:

输入: [1,2,3,0,2]
输出: 3
解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]

解题思路

参考https://blog.csdn.net/weixin_45184581/article/details/120586264

代码

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        if(n==1) return 0;
        //dp[i][0]表示当天持有股票的最大收益
        //dp[i][1]表示当天不持有股票，且当天不是冷静期的最大收益
        //dp[i][2]表示当天不持有股票，且当天是冷静期的最大收益
        int dp[n][3];
        dp[0][0]=-prices[0];
        dp[0][1]=0;
        dp[0][2]=0;
        dp[1][0]=max(dp[0][0],dp[0][1]-prices[1]);
        dp[1][1]=0;
        dp[1][2]=prices[1]-prices[0];
        int i;
        for(i=2;i<n;i++){
            dp[i][0]=dp[i-1][0]>dp[i-1][1]-prices[i]?dp[i-1][0]:dp[i-1][1]-prices[i];
            dp[i][0]=dp[i][0]>dp[i-2][2]-prices[i]?dp[i][0]:dp[i-2][2]-prices[i];

            dp[i][1]=dp[i-1][2]>dp[i-1][1]?dp[i-1][2]:dp[i-1][1];

            dp[i][2]=dp[i-1][0]+prices[i];
        }
        return (dp[n-1][1]>dp[n-1][2]?dp[n-1][1]:dp[n-1][2]);
    }
};