题目链接309. 最佳买卖股票时机含冷冻期
这一题的思路是在原本买卖股票的基础上添加一个冷冻期,即为修改dp条件。先贴出代码:
class Solution {
public:
int maxProfit(vector<int>& prices) {
int dp[prices.size()][2];
dp[0][0]=0;
dp[0][1]=-1*prices[0];
for(int i=1;i<prices.size();i++){
dp[i][0]=max(dp[i-1][0],dp[i-1][1]+prices[i]);
if(i<2){
dp[i][1]=max(dp[i-1][1],dp[i-1][0]-prices[i]);
}
else{
dp[i][1]=max(dp[i-1][1],dp[i-2][0]-prices[i]);
}
}
return dp[prices.size()-1][0];
}
};
首先我们初始化一个二维的dp数组,[0]为当前未买入的最好值,[1]为当前已经被买入的最好值。
首先初始化dp[0][0]为0,即当前处于可买入状态的最优值;
dp[0][1]为-1*prices[0],即为当前可卖出时的最优解。
然后我们逐个进行判断:
首先是 dp[i][0]=max{ dp[i-1][0] , dp[i-1][1] + prices[i] }:
dp[i-1][0]指的是前一天处于未买入状态的最优值,dp[i-1][1]指的是前一天处于已买入状态的最优值。
然后是对i<2的判断
- 当i<2时,此时为第2天(dp[1])
我们考虑当前持股的最优解为第1天(dp[0])为已经持股的最优解和第一天未买入,第二天买入的最优解。 - 当i>=2时,此时需要考虑冷冻期
- 此时持股的最优解为,昨天仍然持股的最优解和前天(及以前)已经卖出(未持股)的最优解的最大值。
所以最后我们只需要考虑最后一天不持股的最优解就是整个过程的最优解了。