代码随想录算法训练营第五十天| 309.最佳买卖股票时机含冷冻期，714.买卖股票的最佳时机含手续费

目录

题目链接：309.最佳买卖股票时机含冷冻期

思路

代码

题目链接：714.买卖股票的最佳时机含手续费

思路

代码

总结

---

题目链接：309.最佳买卖股票时机含冷冻期

思路

        首先明确有几种状态：

0持有；1不持有（昨天之前就是不持有状态和昨天是冷冻期）；2不持有（今天卖）；3冷冻期

                0持有：可能是之前就买入了；或者今天买入，今天买入又分两种，昨天不是冷冻期和昨天是冷冻期

                1不持有：前一天就是保持卖出的状态；昨天是冷冻期

                2不持有：昨天持有，今天卖

                3冷冻期：昨天卖的

        ①dp数组，dp[i][j]表示第i天第j种状态下的最大余额

        ②递推公式
                dp[i][0] = max(dp[i-1][0] , max(dp[i-1][1] - price[i], dp[i-1][3] - price[i]))

                dp[i][1] = max(dp[i-1][1], dp[i-1][3])

                dp[i][2] = dp[i-1][0] + price[i]

                dp[i][3] = dp[i-1][2]

        ③dp数组初始化

                dp[0][0] = -price[0]

                dp[0][1] = 0

                dp[0][2] = 0

                dp[0][3] = 0

        ④遍历顺序，正序遍历，所有状态都有之前的状态推导而来

        ⑤推导dp数组

代码

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int len = prices.size();
        if (len == 1)
            return 0;
        vector<vector<int>> dp(len, vector<int>(4, 0)); // dp数组，四种状态
        dp[0][0] = -prices[0]; // 买入状态的初始化，其他默认0
        for (int i = 1; i < len; i++) {
            // 递推公式
            dp[i][0] = max(dp[i - 1][0], max(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];
        }
        // 最后的结果一定是不持有股票，所以选择除了状态0之外的最大值
        int result = max(dp[len - 1][1], max(dp[len - 1][2], dp[len - 1][3]));
        return result;
    }
};

题目链接：714.买卖股票的最佳时机含手续费

思路

        可无限次交易的股票，增加了手续费，在每次卖出时减去手续费即可。

代码

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

总结

        ①买卖股票问题，不能同时拥有多支股票，分为以下几种：可无限次交易；只能交易一次；只能交易两次；能交易k次；无限次交易但是有冷冻期；无限次交易但是有手续费

        ②这类问题最重要的就是分清楚状态，k次交易和冷冻期的状态需要多看看