算法训练营day53

题目1：309. 买卖股票的最佳时机含冷冻期 - 力扣（LeetCode）

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        vector<vector<int>> dp(prices.size(), vector<int>(3));
        // 三个状态
        // dp[i][0] 持有股票  dp[i][1] 冷冻期  dp[i][2] 不持有股票
        // 递推公式
        // 持有股票 = max(前一次就持有，冷冻期-今天股票价格)
        // 冷冻期 = max(前一次冷冻期，持有股票)
        // 不持有股票 = max(前一次就不持有，持有股票+今天股票价格)
        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][2]);
            dp[i][2] = max(dp[i - 1][2], dp[i - 1][0] + prices[i]);
        }
        return dp[prices.size() - 1][2];
    }
};

题目2：714. 买卖股票的最佳时机含手续费 - 力扣（LeetCode）

这道题就是在 买卖股票Ⅱ上多了一个手续费而已

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