2023.8.21
本题限制了买卖次数:最多买卖两次。 与之前的股票问题相比,dp数组的第二个维度需要设置四个状态:
- 第一次持有股票
- 第一次不持有股票
- 第二次持有股票
- 第二次不持有股票
ps:持有股票不等于购买股票!! 持有股票只是一种状态,可能是之前就买了股票,也可能是刚买入股票,故有两种情况,即:dp[i][0] = max(dp[i-1][0],-prices[i]); 同理,不持有股票也有两种情况,可能是之前就没有,也可能是刚卖出去。即:dp[i][1] = max(dp[i-1][1],prices[i]+dp[i-1][0]); 第二次买入和卖出也是同理。 代码如下:
class Solution {
public:
int maxProfit(vector<int>& prices) {
vector<vector<int>> dp(prices.size(),vector<int>(4));
dp[0][0] = dp[0][2] = -prices[0];
dp[0][1] = dp[0][3] = 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],prices[i]+dp[i-1][0]);
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];
}
};
附上草稿图以供参考: