最佳买卖股票时机含冷冻期(中等)
2020年7月10日
题目来源:力扣
解题
需要好好理解状态量是当前收益,有三种状态:持股状态、不持股的冷冻期状态、不持股的卖出状态
class Solution {
public int maxProfit(int[] prices) {
int plen=prices.length;
int[][] dp=new int[plen+1][3];
if(plen<2) return 0;
//一开始是持股状态
dp[0][0]=-prices[0];
for(int i=1;i<plen;i++){
/*持股状态,第一种情况是上次持股了没卖,继续保持持股状态;
第二种情况是在卖出状态刚买的,进入持股状态*/
dp[i][0]=Math.max(dp[i-1][0],dp[i-1][2]-prices[i]);
/*不持股的冷冻状态,只能是持股卖出且获得收益的*/
dp[i][1]=dp[i-1][0]+prices[i];
/*不持股的卖出状态,第一种情况是不持股也没买,继续保持不持股状态;
第二种情况是刚过冷冻期,进入不持股的卖出状态*/
dp[i][2]=Math.max(dp[i-1][2],dp[i-1][1]);
}
//返回两种不持股状态哪个收益高
return Math.max(dp[plen-1][1],dp[plen-1][2]);
}
}