需要冷却期的股票交易
309. Best Time to Buy and Sell Stock with Cooldown(Medium)
题目描述:
股票交易的原则是先买然后再卖,在买入之前必须至少休息一天,求最后能够获得的最大收益。
思路分析:
动态规划的思想:
sell[i],表示该天结束后手里没有股票的情况下,已经获得的最大收益。
hold[i],表示该天结束后手里有股票的情况下,已经获得的最大收益。
状态转移方程是这样的:
sell[i],表示手里没有股票的收益,这种可能性是今天卖了,或者啥也没干。今天啥也没干那就是sell[i]=sell[i-1],今天卖了,那么sell[i]就是前一天有股票的收益加上今天卖出去股票的价格。sell[i]=hold[i-1]+price[i]
因此:sell[i]=max(sell[i-1],hold[i-1]+price[i])
hold[i],表示今天手里有股票的收益,这种可能性是今天买了股票或者啥也没干。今天啥也没干那就是hold[i]=hold[i-1],今天买了股票,那么hold[i]=sell[i-2]-price[i]
因此:hold[i]=max(hold[i-1],sell[i-2]-price[i])
代码:
public int maxProfit(int []prices){
if(prices==null||prices.length==0)
return 0;
int []sell=new int[prices.length];
int []hold=new int[prices.length];
sell[0]=0;
hold[0]=-prices[0];
for(int i=1;i<prices.length;i++){
sell[i]=Math.max(sell[i-1],hold[i-1]+prices[i]);
hold[i]=Math.max((i>=2?sell[i-2]:0)-prices[i],hold[i-1]);
}
return sell[prices.length-1];
}