Day50 动态规划part11 123. 买卖股票的最佳时机 III 188. 买卖股票的最佳时机 IV

Day50 动态规划part11 123. 买卖股票的最佳时机 III 188. 买卖股票的最佳时机 IV

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

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int length = prices.size(); //dp[i][0]不操作 dp[i][1]第一次持有 dp[i][2]第一次不持有 dp[i][3]第二次持有 dp[i][4]第二次不持有
        if(length==0) return 0;
        vector<vector<int>> dp(length,vector<int>(5,0));

        dp[0][1] = -prices[0];
        dp[0][2] = 0;
        dp[0][3] = -prices[0];
        dp[0][4] = 0;

        for(int i = 1; i< length;i++){
            dp[i][1] = max(dp[i-1][1],-prices[i]);
            dp[i][2] = max(dp[i-1][2],dp[i-1][1]+prices[i]);c
            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[length-1][4];
    }
};

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

运用了空间压缩，编程一维dp状态

class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
        vector<int> dp(2*k+1,0); //dp[i]含义是第(i+1)/2次，如果i%2 == 1那么持有股票，i%2 == 0不持有股票,此时的最大利润
        for(int i = 1; i<=2*k;i++){
            if(i%2 == 1) dp[i] = -prices[0];
        }
        for(int i =1 ;i<prices.size();i++){
            for(int j = 1; j<=2*k-1;j+=2){
                dp[j+1] = max(dp[j+1],dp[j]+prices[i]);
                dp[j] = max(dp[j],dp[j-1]-prices[i]); //奇数天，我们就是第(j+1)/2次持有股票状态，要么维持前一天，要么前一次(j+1)/2-1次持有股票状态，减去当前股价  
            }          
        }
        return dp[2*k];
    }
};