一开始被 至多 买卖两次给唬住,想的是用一个变量表示买卖的次数;
还被 “再次购买前出售掉之前的股票” 给唬住。不知道如何解决
方法:
1. 多用几个下标来表示状态
2. 因为遍历数组是有先后顺序的,在递推公式中也有买卖的先后顺序,所以不用管可能会连续买入两次的情况,因为递推公式里面的推导不会导致这种情况的发生。
与之前不同的是:
有四个下标表示两次的状态 (持有 / 不持有
i 表示天数 , 1 2 3 4表示对应的状态
表示第一次持有股票
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]);
表示第二次不持有股票
dp[i][4] = max(dp[i - 1][4], dp[i - 1][3] + prices[i]);
最后返回的是
dp[prices.size() - 1][4]
因为最后一定是不持有股票的利润最大,而第二次不持有会包含第一次不持有的状态(也就是第一次不持有后,在同一天买入并卖出,这时候第二次不持有和第一次不持有的利润是一样的,
如果还是不理解就取最大值
max ( dp[prices.size() - 1][4] , dp[prices.size() - 1][2] )
class Solution {
public:
int maxProfit(vector<int>& prices) {
/*
:是多用几个下标来表示第一次和第二次分别的状态
*/
vector<vector<int>> dp(prices.size(),vector<int>(5));
dp[0][1] = -prices[0];
dp[0][2] = 0;
dp[0][3] = -prices[0];
dp[0][4] = 0;
for(int i = 1; i < prices.size(); 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]);
dp[i][4] = max(dp[i - 1][4], dp[i - 1][3] + prices[i]);
}
return dp[prices.size() - 1][4];
}
};