309. 最佳买卖股票时机含冷冻期

题目

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

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

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

示例

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

思路

• 每天手里的股票有两种情况，有股票和没股票。用$dp[i][0]$表示第i天手里没股票的最大利润。用$dp[i][1]$表示第i天手里有股票的最大利润。
  • 如果今天手里没股票，那么有两种情况。
  1. 昨天手里没股票。
  2. 昨天手里有股票，今天卖了。
  • 如果今天手里有股票，也有两种情况。
  1. 昨天手里有股票，今天没操作。
  2. 今天买了股票。今天买了股票又分两种情况。a.昨天是冷冻期，即前天卖了股票。b.昨天不是冷冻期，即前天没有卖股票，手里也没股票，昨天没操作。可见应该取a、b中较大利润的选择，而刚好为$dp[i-2][0]$。
• 最后返回值为最后一天手里没股票。

代码

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

        int size = prices.size();

        int dp[size+1][2];
        memset( dp, 0, sizeof(dp) );

        dp[1][1] = -prices[0];

        for ( int i = 2; i <= size; ++i ) {
            dp[i][0] = max( dp[i-1][0], dp[i-1][1] + prices[i-1] );
            dp[i][1] = max( dp[i-1][1], dp[i-2][0] - prices[i-1] );
        }

        return dp[size][0];
    }
};