LeetCode 309. Best Time to Buy and Sell Stock with Cooldown(最佳買賣股票時機含冷凍期)

賣出股票後,你無法在第二天買入股票 (即冷凍期爲 1 天)。

輸入: [1,2,3,0,2]
輸出: 3 
解釋: 對應的交易狀態爲: [買入, 賣出, 冷凍期, 買入, 賣出]
public int maxProfit(int[] prices) {
        if(prices == null) return 0;
        int len = prices.length;
        
        if(len == 0) return 0;

        //定義dp[i][0,1,2]:0表示不持股,1表示持股,2表示處於冷凍期
        int[][] dp = new int[len][3];
        //base case
        dp[0][0] = 0;//不持股
        dp[0][1] =- prices[0];//持股
        dp[0][2] = 0;//冷凍期

        for(int i = 1; i < len; i ++) {
            // 不持股
			// 第i-1天不持股,第i天仍不持股
			// 第i-1天處於冷凍期,第i天就變爲不持股狀態,那麼當前第i天的最大值就是從這兩個狀態中取
            dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][2]);
            // 持股
			// 第i-1天持股,第i天仍持股
			// 第i-1天不持股,第i天買入
            dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][0] - prices[i]);
            // 冷凍期
			// 第i-1天持有股票且今天賣出
            dp[i][2] = dp[i - 1][1] + prices[i];
            
        }
        // 返回不持由股票或者處於冷凍期的最大值
        return Math.max(dp[len - 1][0], dp[len - 1][2]);
    }
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章