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

class Solution {
  public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     * 两次交易所能获得的最大收益
     * @param prices int整型vector 股票每一天的价格
     * @return int整型
     */
    int maxProfit(vector<int>& prices) {
        // write code here
        // 这里我又想起我之前的双指针做法了

        int n = prices.size();
        // 使用n*5的数组保存 状态值  注意初始化为最小！
        vector<vector<int>> dp(n, vector<int>(5, -10001));

        // 边界条件
        dp[0][0] = 0; // 买0 （包括第i天还没买过）
        dp[0][1] = -prices[0]; // 买1 卖0 
        dp[0][2] = 0; // 买1卖1 因为当天买卖 没有收益
        // dp[0][3] = 0; // 买2 卖1  这里不可能  所以不能赋值0 否则就错了
        // dp[0][4] = 0; // 买2卖2

        for(int i=1; i<n; ++i)
        {
            dp[i][0] = dp[i-1][0];
            dp[i][1] = max(dp[i-1][0]-prices[i], dp[i-1][1]);
            dp[i][2] = max(dp[i-1][0], max(dp[i-1][1]+prices[i], dp[i-1][2]));
            dp[i][3] = max(dp[i-1][1], max(dp[i-1][2]-prices[i], dp[i-1][3]));
            dp[i][4] = max(dp[i-1][2], max(dp[i-1][3]+prices[i], dp[i-1][4]));
        }

        
        // int ans = max(dp[n-1][2], max(dp[n-1][3], dp[n-1][4])); // 两者都可
        sort(dp[n-1].begin(), dp[n-1].end());
        int ans = dp[n-1][4];
        return ans>0? ans : 0;
    }
};

看完思路 自己写一遍 注意初始化 和边界条件

O(n) O(n)