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

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

题目链接：Leetcode 309.最佳买卖股票时机含冷冻期
题目描述： 给定一个整数数组prices，其中第 prices[i] 表示第 i 天的股票价格 。​设计一个算法计算出最大利润。在满足以下约束条件下，你可以尽可能地完成更多的交易（多次买卖一支股票），卖出股票后，你无法在第二天买入股票 (即冷冻期为 1 天)。 注意：你不能同时参与多笔交易（你必须在再次购买前出售掉之前的股票）。

思路： 本题的限制条件变成了：卖出股票后一天无法购买股票，因此仅仅按照“持有”和“不持有”股票来分类就不能正确表示状态了。我们还需要将“不持有股票”继续分类：

• 不持有股票且明天不可以购买股票；
• 不持有股票且明天可以购买股票

算上持有股票，总共三种状态。接下来就是确定如何求出递推公式。首先记住这三种状态的含义：

 dp[i][0]: 第i天结束时，持有股票时的最大收益
 dp[i][1]: 第i天结束时，不持有股票且明天不可以购买股票，最大收益
 dp[i][2]: 第i天结束时，不持有股票且明天可以购买股票时，最大收益

• 对于dp[i][0]，可以从dp[i-1][0]转移过来（前一天不进行任何买卖），也可以从dp[i-1][2]转移过来（购买股票）
• 对于dp[i][1]，只能从dp[i-1][0]转移过来（前一天卖掉股票）
• 对于dp[i][2]，可以从dp[i-1][2]转移过来（前一天不进行任何买卖），也可以从dp[i-1][1]转移过来（冷冻期刚好过了）

先理解转移方程的思路，根据这个思路尝试独立写转移方程。写出方程后再思考：这种思路是否完整并且是正确的？ 经过这两步之后会对本题有更深的理解。

接下来该思考如何初始化dp数组：dp[0][0]很好想：-prices[0]，就是第一天买入股票。那dp[0][1]和dp[0][2]呢？尽管第一天不可能出现冷冻期，但是无论如何，我们目前都是不持有股票，因此初始化为0。注：动态规划的很多题目的边界条件没有实际含义或者很难确定其含义，其作用仅仅是为了推出后续dp数组罢了。

代码如下：

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int n = prices.size();
        if (n == 0)
            return 0;
        // dp[i][0]: 第i天结束时，持有股票时的最大收益
        // dp[i][1]: 第i天结束时，不持有股票且明天不可以购买股票，最大收益
        // dp[i][2]: 第i天结束时，不持有股票且明天可以购买股票时，最大收益
        vector<vector<int>> dp(prices.size(), vector<int>(3));
        // 初始化
        dp[0][0] = -prices[0];
        dp[0][1] = dp[0][2] = 0;
        for (int i = 1; i < n; 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[n - 1][1], dp[n - 1][2]); // 无论如何最后手上必须不持有股票
    }
};

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$

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

题目链接：Leetcode 714.买卖股票的最佳时机含手续费
题目描述： 给定一个整数数组 prices，其中 prices[i]表示第 i 天的股票价格 ；整数 fee 代表了交易股票的手续费用。你可以无限次地完成交易，但是你每笔交易都需要付手续费。 如果你已经购买了一个股票，在卖出它之前你就不能再继续购买股票了。返回获得利润的最大值。注意：这里的一笔交易指买入持有并卖出股票的整个过程，每笔交易你只需要为支付一次手续费。

思路： 这道题其实和Leetcode 122.买卖股票的最佳时机II很像，思路与这道题完全相同，仅仅多了手续费而已，我们只需要在卖出股票之后减去手续费即可。

代码如下：

class Solution {
public:
    int maxProfit(vector<int>& prices, int fee) {
        int n = prices.size();
        // dp[i][0]代表持有股票
        // dp[i][1]代表不持有股票
        vector<vector<int>> dp(prices.size(), vector<int>(2));
        // 初始化
        dp[0][0] = -prices[0];
        dp[0][1] = 0;
        for (int i = 1; i < n; 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);//与第122题的唯一区别
        }
        return dp[n - 1][1];//最后一定不持有股票
    }
};

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$

---

总结： 含冷冻期的这道题还是值得仔细思考的，一定要理解好递推公式的思路。

最后，如果文章有错误，请在评论区或私信指出，让我们共同进步！