【LeetCode】714. 买卖股票的最佳时机含手续费

714. 买卖股票的最佳时机含手续费（中等）

思路

股票相关的题都可以建立状态机，很快就能表示出状态之间的变化。

对于本题，只有两种状态，买入（持有）或卖出（不持有），只需要对这两种状态进行定义。

状态定义

• buy[i]: 第 i 天买入（持有）股票时的最大利润；
• sell[i]：第 i 天卖出（不持有）股票时的最大利润。

状态转移方程

对于 buy 和 sell ，都有两种状态转移的情况：

首先讨论 buy：

• 在「持有」股票（之前已经买入）的情况下，继续保持「持有」状态，那么当前的最大收益和之前相同，buy[i] = buy[i-1]；
• 上一个状态是「不持有」（之前已经卖出）的情况下，买入股票变成「持有」状态，那么当前的最大收益就是「不持有」股票的最大收益再减去当前股票的价格，即 buy[i] = sell[i-1] - prices[i]；
• 要想获得最大利益，就是选取两种情况的最大值，即 buy[i] = max(buy[i-1], sell[i-1] - prices[i]);

接下来讨论 sell：

• 在「不持有」股票（之前已经卖出）的情况下，继续保持「不持有」状态，那么当前的最大收益和之前相同，sell[i] = sell[i-1]；
• 上一个状态是「持有」（之前已经买入）的情况下，卖出股票变成「不持有」状态，完成一次交易，需要付出手续费，那么当前的最大收益就是「持有」股票的最大收益再加上当前股票的价格，并减去手续费，即 sell[i] = buy[i-1] + prices[i] - fee；
• 要想获得最大利益，就是选取两种情况的最大值，即 sell[i] = max(sell[i-1], buy[i-1] + prices[i] - fee);

代码

class Solution {
public:
    int maxProfit(vector<int>& prices, int fee) {
        int n = prices.size();
        vector<int> buy(n), sell(n);
        
        // 初始化
        buy[0] = -prices[0], sell[0] = 0;

        for(int i=1; i<n; ++i){
            buy[i] = max(buy[i-1], sell[i-1] - prices[i]);
            sell[i] = max(sell[i-1], buy[i-1] + prices[i] - fee);
        }
        return sell[n-1];
    }
};

初始化

• 对于 buy[0]，在第 0 天的时候要持有股票，就需要买入价格为 prices[0] 的股票，此时的最大收益为：buy[0] = -prices[0];
• 对于 sell[0] ，第 0 天的时候不持有股票，可以选择不进行任何操作，也可以选择当天买入又售出，但是后者的话需要付出手续费，得不偿失，因此不进行任何操作能确保收益更大：sell[0] = 0。

最终的返回结果

最后一天卖出（不持有）股票的最大收益： sell[n-1] ，就是利润的最大值。

优化：空间压缩

显然，buy 和 sell 的当前状态都只和上一个状态有关，因此可以用两个变量来表示该状态。

代码

class Solution {
public:
    int maxProfit(vector<int>& prices, int fee) {
        int n = prices.size();
        
        // 初始化
        int buy = -prices[0], sell = 0;

        for(int i=1; i<n; ++i){
            buy = max(buy, sell - prices[i]);
            sell = max(sell, buy + prices[i] - fee);
        }
        return sell;
    }
};