代码随想录训练营第五十一天

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

把不持有股票状态拆分成两个状态：①保持卖出股票 ②卖出股（具体卖出股票的动作状态）

为什么要拆分？因为冷冻期的前一个状态必然是具体卖出股票的那一个状态，而不是简单的不持有股票。

递推公式

初始化

不合理的状态一般默认为0

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if(prices.size() == 0){
            return 0;
        }
        vector<vector<int>> dp(prices.size(),vector<int>(4,0));
        dp[0][0] = -prices[0];
        for(int i = 1;i < prices.size();i++){
            dp[i][0] = max({(int)dp[i-1][0],dp[i-1][1]-prices[i],dp[i-1][3]-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({(int)dp[prices.size()-1][1],dp[prices.size()-1][2],dp[prices.size()-1][3]});
    }
};

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

就是在122题基础上卖出股票的时候减去手续费就可。

class Solution {
public:
    int maxProfit(vector<int>& prices, int fee) {
        int len = prices.size();
        if(len == 0){
            return 0;
        }
        vector<vector<int>> dp(len, vector<int>(2));
        dp[0][0] = - prices[0];
        dp[0][1] = 0;
        for(int i = 1;i < len;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[len-1][1];
    }
};