代码随想录算法训练营第五十天|123.买卖股票的最佳时机III、188.买卖股票的最佳时机IV

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

文章链接：代码随想录

视频链接：LeetCode:123.买卖股票的最佳时机3

题目链接：力扣题目链接

图释：

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int len = prices.size();
        if(len==0) return 0;
        //dp[]数组的含义：dp[i][0]不操作
        // dp[i][1]:第一次持有  dp[i][2]:第一次不持有
        // dp[i][3]:第二次持有  dp[i][4]:第二次不持有
        vector<vector<int>> dp(len, vector<int>(5));
        // 初始化
        dp[0][0] = 0;
        dp[0][1] = -prices[0]; 
        dp[0][2] = 0;
        dp[0][3] = -prices[0]; 
        dp[0][4] = 0;

        for(int i=1; i<len; i++){
            dp[i][1] = max(dp[i-1][1], -prices[i]); //延续第i-1天持有的状态，或者当天购买
            dp[i][2] = max(dp[i-1][2], dp[i-1][1]+prices[i]); //延续第i-1天不持有的状态，或者当天卖出
            dp[i][3] = max(dp[i-1][3], dp[i-1][2]-prices[i]); //延续第i-1天第二次持有的状态，或者当天第二次购买
            dp[i][4] = max(dp[i-1][4], dp[i-1][3]+prices[i]); //延续第i-1天第二次不持有的状态，或者当天第二次卖出
        }
        return dp[len-1][4]; //如果是只买一次就有就有最大值了话，可以同一天买卖出
    }
};

题目：188.买卖股票的最佳时机IV

文章链接：代码随想录

视频链接：LeetCode:188.买卖股票的最佳时机4

题目链接：力扣题目链接

图释：

class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        int len = prices.size();
        if(len==0) return 0;
        //dp[]数组的含义：dp[i][0]不操作
        // dp[i][1]:第一次持有  dp[i][2]:第一次不持有 同理可推，奇数为持有，偶数为不持有
        vector<vector<int>> dp(len, vector<int>(2*k+1, 0));
        // 初始化
        for(int j=1; j<=2*k; j+=2){ //去奇数
           dp[0][j] = -prices[0]; //第一天持有的状态，不管买卖几次
        }
    
        for(int i=1; i<len; i++){
            for(int j=1; j<2*k+1; j+=2){
                dp[i][j] = max(dp[i-1][j], dp[i-1][j-1]-prices[i]); //延续第i-1天持有的状态，或者当天购买
                dp[i][j+1] = max(dp[i-1][j+1], dp[i-1][j]+prices[i]); //延续第i-1天不持有的状态，或者当天卖出
            }
        }
        return dp[len-1][2*k]; //如果是只买一次就有就有最大值了话，可以同一天买卖出
    }
};