本题关键在于至多买卖两次,这意味着可以买卖一次,可以买卖两次,也可以不买卖。
那么需要在买一次的状态上,增加买第二次的状态的判断,状态有以下四种:
- 第一次持有股票
- 第一次不持有股票
- 第二次持有股票
- 第二次不持有股票
-
dp[i][0] – dp[i][3] 分别表示上述状态时获得的最大现金
-
操作一:第i天买入股票了,那么dp[i][0] = - prices[i]
操作二:第i天没有操作,而是沿用前一天买入的状态,即:dp[i][0] = dp[i - 1][0]
dp 数组的值 为每天的最大现金,因此取两者的最大值 dp[i][0] = max(dp[i - 1][0], - prices[i]);
同理:
dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i]);
dp[i][2] = max(dp[i - 1][2], dp[i - 1][1] - prices[i]);
dp[i][3] = max(dp[i - 1][3], dp[i - 1][2] + prices[i]); -
第0天做第一次买入的操作,dp[0][0] = -prices[0];
第0天做第一次出售操作,dp[0][1] = 0;
第0天做第二次买入操作, dp[0][2] = - prices[0];
第0天做第二次出售操作,dp[0][3] = 0;
class Solution {
public:
int maxProfit(vector<int>& prices) {
vector<vector<int>> dp(prices.size(), vector<int>(4, 0));
dp[0][0] = -prices[0];
dp[0][2] = -prices[0];
for(int i = 1; i < prices.size(); i++){
dp[i][0] = max(dp[i - 1][0], - prices[i]);
dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i]);
dp[i][2] = max(dp[i - 1][2], dp[i - 1][1] - prices[i]);
dp[i][3] = max(dp[i - 1][3], dp[i - 1][2] + prices[i]);
}
return dp[prices.size() - 1][3];
}
};