第四十二天 第九章 动态规划part09 188.买卖股票的最佳时机IV 309.最佳买卖股票时机含冷冻期 714.买卖股票的最佳时机含手续费

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

这题和上面一题很像，但是我们的交易次数变为k。

我们只要分清楚奇数表示我们为买入状态，偶数表示我们为卖出状态。

dp[i][1]，表示的是第i天，买入股票的状态，并不是说一定要第i天买入股票，可能是前一天已经买入了。他只是表现为一个状态。

class Solution {
public:
    int maxProfit(int k, vector<int>& prices) {
      vector<vector<int>> dp(prices.size(),vector<int>(2*k+1,0));  
      for(int i=1;i<k*2;i+=2){
        dp[0][i]=-prices[0];
      }
      for(int i=1;i<prices.size();i++){
        for(int j=0;j<k*2;j+=2){
            dp[i][j+1]=max(dp[i-1][j+1],dp[i-1][j]-prices[i]);
            dp[i][j+2]=max(dp[i-1][j+2],dp[i-1][j+1]+prices[i]);
        }

    }
     return dp[prices.size()-1][2*k];
    }
};

309.最佳买卖股票时机含冷冻期  

代码随想录里面分为了四个状态，其实三个状态也可以。

初始化可以带入递推公式去计算。

四个状态：把不持有股票分为冷冻期和过了冷冻期并且不持有股票两种

class Solution {
public:
    int maxProfit(vector<int>& prices) {
    vector<vector<int>> dp(prices.size()+1,vector<int>(4));
     dp[0][0] = -prices[0];
     for(int i=1;i<prices.size();i++){
        dp[i][0]=max(dp[i-1][0], max(dp[i-1][3], dp[i-1][1]) - prices[i]);   //持有股票
        dp[i][1]=max(dp[i-1][1],dp[i-1][3]);  //保持卖出股票(也就是冷冻期后一直没持有的状态)
        dp[i][2]=dp[i-1][0]+prices[i]; //卖出股票
        dp[i][3]=dp[i-1][2]; // 冷冻期
     }
     return max(dp[prices.size()-1][3], max(dp[prices.size()-1 ][1], dp[prices.size()-1][2]));
    }
};

 三种状态：

有两个方法

第一种方法有点不能理解。

第二种方法相当于把过了冷冻期并且不持有股票的状态看作是冷冻期。

class Solution {
public:
    int maxProfit(vector<int>& prices) {
    vector<vector<int>> dp(prices.size()+1,vector<int>(3));
     dp[0][0] = -prices[0];
     for(int i=1;i<prices.size();i++){
        dp[i][0]=max(dp[i-1][0], dp[i-1][2] - prices[i]);   //持有股票
        dp[i][1]=max(dp[i-1][1],dp[i-1][0]+prices[i]); //卖出股票
        dp[i][2]=dp[i-1][1]; // 冷冻期
     }
     return max(dp[prices.size()-1 ][1], dp[prices.size()-1][2]);
    }
};


class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if (prices.empty()) return 0;
        vector<vector<int>> dp(prices.size(), vector<int>(3, 0));
        dp[0][0] = -prices[0];
        for (int i = 1; i < prices.size(); i++) {
            dp[i][0] = max(dp[i-1][0], dp[i-1][2] - prices[i]);  // 持有股票
            dp[i][1] = dp[i-1][0] + prices[i];                   // 卖出股票
            dp[i][2] = max(dp[i-1][1], dp[i-1][2]);              // 冷冻期
        }
        return max(dp[prices.size() - 1][1], dp[prices.size() - 1][2]);
    }
};

714.买卖股票的最佳时机含手续费  

和上面题的区别在于需要手续费。

class Solution {
public:
    int maxProfit(vector<int>& prices, int fee) {
        vector<vector<int>> dp(prices.size()+1,vector<int>(2));
        dp[0][0]=-prices[0];
        for(int i=1;i<prices.size();i++){
            dp[i][0]=max(dp[i-1][0],dp[i-1][1]-prices[i]);
            dp[i][1]=max(dp[i-1][1],dp[i-1][0] + prices[i]-fee);
        }
        return dp[prices.size()-1][1];
    }
};