题解 | 买卖股票的最好时机(三)

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 两次交易所能获得的最大收益
     * @param prices int整型vector 股票每一天的价格
     * @return int整型
     */
    int maxProfit(vector<int>& prices) {
        // write code here
        int n = prices.size();
        vector<vector<int>> dp(5, vector<int>(n,
                                              INT_MIN / 2));
                                              
        dp[0][0] = 0;
        dp[1][0] = -prices[0];
        for(int j=1;j<n;j++){
            dp[0][j] = dp[0][j-1];
            dp[1][j] = max(dp[1][j-1],dp[0][j-1]-prices[j]);
            dp[2][j] = max(dp[2][j-1],dp[1][j-1]+prices[j]);
            dp[3][j] = max(dp[3][j-1],dp[2][j-1]-prices[j]);
            dp[4][j] = max(dp[4][j-1],dp[3][j-1]+prices[j]);
        }
        return max({dp[2][n-1],0,dp[4][n-1]});
    }
};