123. 买卖股票的最佳时机 III

✅做题思路or感想

考验的是对二维dp数组的理解和运用，一维是天数，二维是状态，了解了就很好做了

dp数组含义

dp[i][0]：第一次交易持有股票能获得的最大利润

dp[i][1]：第一次交易没有股票能获得的最大利润

dp[i][2]：第二次交易持有股票能获得的最大利润

dp[i][3]：第二次交易没有股票能获得的最大利润

推导公式

dp[i][0]可能是前一天就持有股票，可能是今天才买，这里因为是第一次购买股票，所以获得的利润一定是-prices[i]，故有dp[i][0] = max(dp[i - 1][0], -prices[i])

dp[i][1]可能是前一天就没有股票，也可能是今天才卖掉股票，故dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i])

dp[i][2]可能是是前一天就有股票，可能是今天才买，因为这里是第二次购买，所以它利用的前一天没买的数据应该是dp[i][1]，故有dp[i][2] = max(dp[i - 1][2], dp[i - 1][1] - prices[i])

dp[i][3]可能是前一天就没有股票，也可能是今天才把股票卖了，故有dp[i][3] = max(dp[i - 1][3], dp[i - 1][2] + prices[i])

初始化

dp[0][0] = -prices[0];	//第一天就买了股票
dp[0][1] = 0;	//第一天什么也没干
dp[0][2] = -prices[0];	//第一条买了又卖了后，又买了
dp[0][3] = 0;	//第一天买了又卖了，然后又买了又卖了

遍历顺序

从小推大，所以是正序

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        vector<vector<int>>dp (prices.size() + 1, vector<int>(4, 0));
        //0是第一次交易持有股票，1是第一次交易没有股票
        //2是第二次交易持有股票，3是第二次交易没有股票
        dp[0][0] = -prices[0];
        dp[0][1] = 0;
        dp[0][2] = -prices[0];
        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], 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];
    }
};