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

309. 最佳买卖股票时机含冷冻期
     给定一个整数数组，其中第 i 个元素代表了第 i 天的股票价格 。​

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

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

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

题解

比较明显的动态规划，dp[i]表示区间[0,i]之间按照题目要求得到的最大收益，那么更新过程就是
dp[i]=max(dp[i],dp[j-2]+get_max_profit(j,i));
其中get_max_profit(j,i)表示在区间[j,i]进行一次买卖获得的最大收益，因为卖之后至少需要一天的冷冻期才能再买，所以不能是dp[j-1]只能是dp[j-2]中间跳过一天。

AC代码

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