714.买卖股票的最佳时机Ⅵ

题目描述

题解

动态规划

虽然增加了交易费用, 但是对于整体的算法思路没有影响, 简单地将其归入买入价格的一部分即可

即dp[i][1]的值为Math.max(dp[i - 1][1], dp[i - 1][0] - prices[i] - fee)

同时要注意初始化时dp[0][1]的值为-prices[0] - fee

public int maxProfit(int[] prices, int fee) {
    int len = prices.length;
    if (len < 2) {
        return 0;

    }

    int[][] dp = new int[len][2];
    dp[0][0] = 0;
    dp[0][1] = -prices[0] - fee;

    for (int i = 1; i < len; i++) {
        dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][1] + prices[i]);
        dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][0] - prices[i] - fee);
    }
    return dp[len - 1][0];
}