给定一个整数数组prices,其中第 prices[i] 表示第 i 天的股票价格 。
设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):
卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
含有冷冻期,就在每天里加上冷冻期的状态。有三种状态:未持有、持有、冷冻期
未持有的由来:延续上一天的未持有/上一天持有今天卖了
持有的由来:延续上一天的持有/上一天是冷冻期今天又买了
冷冻期的由来:只能是两天前持有,一天前卖了(还是得和冷冻期的前一天状态比较)
另外,在初始化时,由于冷冻期的状态与两天前有关,所以初始化得初始化两天。
class Solution {
public:
int maxProfit(vector<int>& prices) {
//每天有未持有、持有、冷冻期三个状态
vector<vector<int>> dp(prices.size(),vector<int>(3,0));
if(prices.size()==1) return 0;
dp[0][0]=0;
dp[0][1]=-prices[0];
dp[0][2]=0;
dp[1][0]=max(0,prices[1]-prices[0]);
dp[1][1]=max(-prices[0],-prices[1]);
dp[1][2]=0;
for(int i=2;i<prices.size();i++){
//未持有
dp[i][0]=max(dp[i-1][0],dp[i-1][1]+prices[i]);
//持有
dp[i][1]=max(dp[i-1][1],dp[i-1][2]-prices[i]);
//冷冻期
dp[i][2]=max(dp[i-1][2],dp[i-2][1]+prices[i-1]);
}
return max(dp[prices.size()-1][0],dp[prices.size()-1][2]);
}
};