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

题目： 给定一个整数数组prices，其中第 prices[i] 表示第 i 天的股票价格 。​设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票）:

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

示例 1:

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

示例 2:

输入: prices = [1]
输出: 0

思路：

dp[i][0] 持股
dp[i][1] 保持卖出股
dp[i][2] 卖出股
dp[i][3] 冷冻期
保持冷冻期就是在冷冻期后的一段时间，还没有买入新的股票的时候
冷冻期是卖出股的后一天
递推公式：
dp[i][0]可以由什么推导出来：昨天已经持股了；昨天是冷冻期今天买了；昨天是保持冷冻期今天买了
dp[i - 1][0]
dp[i - 1][3] - prices[i]
dp[i - 1][1] - prices[i]
dp[i][1]可以由什么推导出来：上一天已经是卖出股了；上一天是冷冻期
dp[i - 1][1]
dp[i - 1][3]
dp[i][2]可以由什么推导出来：只能由昨天持股推导出来
dp[i - 1][0] + prices[i]
dp[i][3]可以由什么推导出来：只能由昨天卖出股推导出来
dp[i - 1][2]
初始化：
dp[0][0] = -prices[0]
dp[0][1] = 0
dp[0][2] = 0
dp[0][3] = 0
顺序：从小到大

class Solution {
public:
	int maxProfit(vector<int>& prices) {
		if (prices.size() == 0)return 0;
		vector<vector<int>> dp(prices.size()+1,vector<int>(4));
		dp[0][0] = -prices[0];
		for (int i = 1; i < prices.size();i++) {
			dp[i][0] = max(max(dp[i - 1][0], dp[i - 1][3] - prices[i]), 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[prices.size() - 1][3], max(dp[prices.size() - 1][1], dp[prices.size() - 1][2]));
	}
};

int main() {
	vector<int> prices = { 1, 2, 3, 0, 2 };
	Solution ss;
	cout<<ss.maxProfit(prices)<<endl;

	return 0;
}