1. 程式人生 > >LeetCode-【動態規劃】-最佳買賣股票時機含冷凍期

LeetCode-【動態規劃】-最佳買賣股票時機含冷凍期

給定一個整數陣列,其中第 i 個元素代表了第 i 天的股票價格 。​

設計一個演算法計算出最大利潤。在滿足以下約束條件下,你可以儘可能地完成更多的交易(多次買賣一支股票):

  • 你不能同時參與多筆交易(你必須在再次購買前出售掉之前的股票)。
  • 賣出股票後,你無法在第二天買入股票 (即冷凍期為 1 天)。

示例:

輸入: [1,2,3,0,2]
輸出: 3 
解釋: 對應的交易狀態為: [買入, 賣出, 冷凍期, 買入, 賣出]

題解:對任意一天共有兩種狀態存在,一是持有股票,二是不持有股票(包括冷凍期)。

持有股票:

1.今天買入 2.前一天沒有賣出 。最終最大利潤去這兩中情況的最大值。

未持有股票

1.和兩天前一樣未持有(包括冷凍期)2.今天賣出。最終最大利潤去這兩中情況的最大值。

根據以上分析,分別建立持有股票陣列h[n],dh[n],其中n是天數,那麼有

h[i]=max(h[i-1],dh[i-2]-prices[i]);

dh[i]=max(dh[i-1],h[i-1]+prices[i]);

class Solution {
    public int maxProfit(int[] prices) {
        int n=prices.length;
        if(n==0)
            return 0;
        int[] dh=new int[n];
        int[] h=new int[n];
        dh[0]=0;
        h[0]=-prices[0];
        for(int i=1;i<n;i++){
            dh[i]=Math.max(dh[i-1],h[i-1]+prices[i]);
            if(i>=2)
                h[i]=Math.max(h[i-1],dh[i-2]-prices[i]);
            else
                h[i]=Math.max(h[i-1],-prices[i]);
        }
        return dh[n-1];
    }
}