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

解答

动态规划

第122题的变形，只需要考虑买入股票时减去手续费即可

#include <vector>
#include <algorithm>
using namespace std;

class Solution {
public:
    int maxProfit(vector<int>& prices, int fee) {
        int n = prices.size();
        if(n<2) return 0;

        vector<vector<int>> dp(n,vector<int>(2));

        //手上没有股票，收益为0
        dp[0][0] = 0;

        // 手上持有了 1 份股票，那肯定是执行了买入操作，然后又还没有卖出，那么钱都投入了股票中，利润就是负的，即为 -prices[0]
        // 由于每笔交易你只需要为支付一次手续费，因此这里可以认为是买入的时候就支付了手续费，
        // 那么此时的利润是 - prices[0] - fee)
        dp[0][1] = -prices[0]-fee;

        for (int i = 1; i < n; i++)
        {
            // 1、今天【不持有】股票
            // 第 i - 1 天【持有】股票，第 i 天卖出
            // 昨天【持有】股票，今天卖出，不需要支付手续费
            // vs
            // 第 i - 1 天【不持有】股票，第 i 天不操作
            // 昨天【不持有】股票，今天不操作
            dp[i][0] = max(dp[i-1][0],dp[i-1][1]+prices[i]);

            // 2、今天【持有】股票
            // 第 i - 1 天【持有】股票，第 i 天不操作
            // 昨天【持有】股票，今天不操作
            // vs
            // 第 i - 1 天【不持有】股票，第 i 天买入
            // 昨天【不持有】股票，今天买入，需要支付手续费
            dp[i][1] = max(dp[i-1][1], dp[i-1][0]-prices[i]-fee);
        }
        // 表示第 n - 1 天结束时，即收盘后，手上持有 0 份股票，可以获得的最大收益
        return dp[n-1][0];
    }
};