算法修炼Day51|● 309.最佳买卖股票时机含冷冻期 ● 714.买卖股票的最佳时机含手续费

51 阅读1分钟
题目:309. 买卖股票的最佳时机含冷冻期 - 力扣(LeetCode)
代码实现:
class Solution {
    public int maxProfit(int[] prices) {
        if (prices.length < 2) return 0;
        int[][] dp = new int[prices.length][3];
        dp[0][0] = 0;
        dp[0][1] = -prices[0];
        dp[0][2] = 0;
        // dp[i][0]:不持有股票,且今天不是由于卖出股票而不持股
        // dp[i][1]:持有股票
        // dp[i][2]:不持有股票,且今天是由于卖出股票而不持股
        for (int i = 1; i < prices.length; i++) {
            dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][2]); // 不持有:前一天不持有(包括卖出和不卖出两种状态)
            dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][0] - prices[i]); // 持有:前一天就持有,前一天不持有且不是因为卖出不持有且当天买入持有
            dp[i][2] = dp[i - 1][1] + prices[i]; // 前一天持有且当天卖出
        }
        return Math.max(dp[prices.length - 1][0], dp[prices.length - 1][2]);
    }
}
题目:714. 买卖股票的最佳时机含手续费 - 力扣(LeetCode)
代码实现:
class Solution {
    public int maxProfit(int[] prices, int fee) {
        // 两种状态:持有 或 不持有
        int[][] dp = new int[prices.length][2];
        dp[0][0] = -prices[0];
        for (int i = 1; i < prices.length; 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 Math.max(dp[prices.length - 1][1], dp[prices.length - 1][0]);
    }
}