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

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

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

思路：本题难点在于如何将冷冻期加入到状态转移方程中，不妨画个图：

按理来说，如何我们正处于买入状态，将股票卖出后，应该是冷冻状态，但是这里多加了一个今日卖出状态，就是将今日卖出和卖出状态分开看，避免两个状态（买入态和冷动态）之间相互转化，否则容易把自己绕进去了。

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

好像三个状态也挺好理解的

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        if (prices.empty()) {
            return 0;
        }

        int n = prices.size();
        // f[i][0]: 手上持有股票的最大收益
        // f[i][1]: 手上不持有股票，并且处于冷冻期中的累计最大收益
        // f[i][2]: 手上不持有股票，并且不在冷冻期中的累计最大收益
        vector<vector<int>> f(n, vector<int>(3));
        f[0][0] = -prices[0];
        for (int i = 1; i < n; ++i) {
            f[i][0] = max(f[i - 1][0], f[i - 1][2] - prices[i]);
            f[i][1] = f[i - 1][0] + prices[i];
            f[i][2] = max(f[i - 1][1], f[i - 1][2]);
        }
        return max(f[n - 1][1], f[n - 1][2]);
    }
};

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

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

思路：别忘了初始化的时候也有要收手续费

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

总结

代码随想录 (programmercarl.com)