算法训练营第四十九天|121. 买卖股票的最佳时机、122.买卖股票的最佳时机II

47 阅读1分钟

121. 买卖股票的最佳时机

题目规定只能买卖一次,所以这时每一个第i天买入股票的话,就是只能在这天才是第一次买,所以Math.max比较的对象是-prices[i]而不是dp[i - 1][1] - prices[i]

class Solution {
    public int maxProfit(int[] prices) {
        // dp[i][0]:第i天持有股票的最大利润
        // dp[i][1]:第i天持有现金的最大利润
        int[][] dp = new int[prices.length][2];

        dp[0][0] = -prices[0];
        dp[0][1] = 0;

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

        return dp[prices.length - 1][1];
    }
}

122. 买卖股票的最佳时机 II

这题规定是可以多次买卖股票,所以就是在dp[i][0]的推导上会有所不同,会有对前一天持有现金的利润的依赖,即为dp[i - 1][1] - prices[i]

class Solution {
    public int maxProfit(int[] prices) {
        // dp[i][0]:第i天持有股票的最大利润
        // dp[i][1]:第i天持有现金的最大利润
        int[][] dp = new int[prices.length][2];

        dp[0][0] = -prices[0];
        dp[0][1] = 0;

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

        return dp[prices.length - 1][1];
    }
}