LeetCode-【動態規劃】-最佳買賣股票時機含冷凍期
阿新 • • 發佈:2019-01-26
給定一個整數陣列,其中第 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]; } }