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

给定一个整数数组 prices，其中第 i 个元素代表了第 i 天的股票价格 ；非负整数 fee 代表了交易股票的手续费用。

你可以无限次地完成交易，但是你每笔交易都需要付手续费。如果你已经购买了一个股票，在卖出它之前你就不能再继续购买股票了。

返回获得利润的最大值。

注意：这里的一笔交易指买入持有并卖出股票的整个过程，每笔交易你只需要为支付一次手续费。

示例 1:

输入: prices = [1, 3, 2, 8, 4, 9], fee = 2
输出: 8
解释: 能够达到的最大利润:  
在此处买入 prices[0] = 1
在此处卖出 prices[3] = 8
在此处买入 prices[4] = 4
在此处卖出 prices[5] = 9
总利润: ((8 - 1) - 2) + ((9 - 4) - 2) = 8.

注意:

0 < prices.length <= 50000.
0 < prices[i] < 50000.
0 <= fee < 50000.

解答

这道题和基本的买卖股票 2唯一差异在于每次交易都需要支付手续费，所以需要减少交易次数，但是仍然可以延续此前贪心的策略。
首先考虑prices=[1, 4, 7], fee=2这样的例子，按照此前的贪心策略的话就是每一步都执行交易，收益为4-1-2+7-4-2=2，但考虑手续费的话应该只在直接在1处购入，7处卖出，此时收益为7-1-2=4，沿用贪心思路，我们依然可以选择只要当前价格比此前价格高就进行交易，但如果连续的两次交易发生在同一个价格处（上述例子的价格4)，那显然可以减少一次交易，转换一下思路就是在该处售出股票后，将其价格相应减去fee，这样如果沿用贪心策略发生在售出处再次执行购入的话，就等价于减少了一次交易。

class Solution {
public:
    int maxProfit(vector<int>& prices, int fee) {
        int result = 0;
        int pre = prices[0];
        for(int i = 0; i < prices.size(); i++){
            if(prices[i] < pre){
                pre = prices[i];
            }
            // 如果 pre 减去了 fee，那只要价格上升就可以交易
            // 如果 pre 没有减 fee，那必须要大于购入价格+fee则交易
            else if(prices[i] > (pre + fee)){
                result += (prices[i] - pre - fee);
                pre = prices[i] - fee;
            }
        }
        return result;
    }
};

同样可以采用动态规划，dp[i][0]表示i位置没有股票，dp[i][1]表示持有股票：

class Solution {
public:
    int maxProfit(vector<int>& prices, int fee) {
        vector<vector<int>> dp(prices.size(), vector<int>(2, 0));
        dp[0][1] -= prices[0];
        for(int i = 1; i < prices.size(); i++){
            dp[i][0] = max(dp[i - 1][0], dp[i - 1][1] + prices[i] - fee);
            dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] - prices[i]);
        }
        return dp[prices.size() - 1][0];
    }
};