买卖股票的最佳时机含冷冻期 困难🌟🌟
问题描述
原文链接：309. 买卖股票的最佳时机含冷冻期
给定一个整数数组prices，其中第 prices[i] 表示第 *i* 天的股票价格 。
设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票）:
- 卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。
注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。
示例 1:
输入: prices = [1,2,3,0,2]
输出: 3
解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]
示例 2:
输入: prices = [1]
输出: 0
提示：
1 <= prices.length <= 50000 <= prices[i] <= 1000
代码实现
Java
class Solution {
/*
1、dp[i][j]：表示在第 i 天结束后，股票的状态为 j，此时我们手上拥有的最大现金为 dp[i][j]。
j = 0:表示不持有股票且不处于冷却期，（代表第 i+1 可以买卖股票）
j = 1:表示不持有股票且处于冷却期，（代表第 i+1 不可以买卖股票）
j = 2:表示我们持有股票，此时我们手上拥有的最大现金为 dp[i][j]
最终要求的：dp[n-1][0]。
2、dp[i][0]:表示不持有股票且不处于冷却期
a.说明今天肯定不能卖出股票：说明，昨天不持有股票
dp[i][0] = Math.max(dp[i-1][0], dp[i-1][1]);
dp[i][1]:表示不持有股票且处于冷却期
a.说明今天卖出了股票 dp[i][1] = dp[i- 1][2] + prices[i];
dp[i][2]:表示我们持有股票
a.今天什么也没做，说明昨天就拥有了股票：dp[i][2] = dp[i-1][2]
b.今天买进这个股票，说明昨天是没有股票的：dp[i][2] = dp[i-1][0] - prices[i]
dp[i][2] = Math.max(dp[i-1][2], dp[i-1][0] - prices[i]);
3、初始值
dp[0][0] = 0;
dp[0][1] = 0;
dp[0][2] = -prices[0];
*/
public int maxProfit(int[] prices) {
int n = prices.length;
int[][] dp = new int[n][3];
dp[0][2] = -prices[0];
for(int i = 1; i < n; i++){
dp[i][0] = Math.max(dp[i-1][0], dp[i-1][1]);
dp[i][1] = dp[i- 1][2] + prices[i];
dp[i][2] = Math.max(dp[i-1][2], dp[i-1][0] - prices[i]);
}
return Math.max(dp[n-1][0], dp[n-1][1]);
}
}
Python
n = len(prices)
dp = [[0] * 3 for _ in range(n)]
dp[0][2] = -prices[0]
for i in range(1, n):
dp[i][0] = max(dp[i-1][0], dp[i-1][1])
dp[i][1] = dp[i-1][2] + prices[i]
dp[i][2] = max(dp[i-1][2], dp[i-1][0] - prices[i])
return max(dp[n-1][0], dp[n-1][1])
C++
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n = prices.size();
vector<vector<int>> dp(n, vector<int>(3));
dp[0][2] = -prices[0];
for(int i = 1; i < n; i++){
dp[i][0] = max(dp[i-1][0], dp[i-1][1]);
dp[i][1] = dp[i-1][2] + prices[i];
dp[i][2] = max(dp[i-1][2], dp[i-1][0] - prices[i]);
}
return max(dp[n-1][0], dp[n-1][1]);
}
};
Go
func maxProfit(prices []int) int {
n := len(prices)
dp := make([][]int, n)
for i := range dp {
dp[i] = make([]int, 3)
}
dp[0][2] = -prices[0]
for i := 1; i < n; i++ {
dp[i][0] = max(dp[i-1][0], dp[i-1][1])
dp[i][1] = dp[i-1][2] + prices[i]
dp[i][2] = max(dp[i-1][2], dp[i-1][0] - prices[i])
}
return max(dp[n-1][0], dp[n-1][1])
}
func max(a, b int) int {
if a > b {
return a
}
return b
}