309. 最佳买卖股票时机含冷冻期（中等）

思路：

动态规划dp

 

代码：

class Solution {
    public int maxProfit(int[] prices) {
		int n=prices.length;
		if(n<2) return 0;
		int[][] dp=new int[n][3];
		
		//初始化
		//0:不持股，不在冰冻期
		//1:持股
		//2：不持股，在冰冻期
		dp[0][0]=0;
		dp[0][1]=-prices[0];
		dp[0][2]=0;
		
		//股票卖掉后进入冰冻期
		for(int i=1;i<n;i++){
			//不在冰冻期：前一个不在冰冻期/前一个就在冰冻期，现在这个变为不冰冻期
			dp[i][0]=Math.max(dp[i-1][0],dp[i-1][2]);
			//持股：前一个就持股/前一个不持股并且不在冰冻期，现在买入了股票
			dp[i][1]=Math.max(dp[i-1][1],dp[i-1][0]-prices[i]);
			//不持股在冰冻期：前一个持股卖掉了。不可能前一个还是冰冻期，冰冻期只有1天
			dp[i][2]=dp[i-1][1]+prices[i];
			
		}
		//比较不持股：在冰冻期/不在冰冻期，哪一个更多钱。不比较还持股的
		return Math.max(dp[n-1][0],dp[n-1][2]);
	}
}

 

分解：

1）三种状态：

0:不持股，不在冰冻期
1:持股
2：不持股，在冰冻期

注意：冷冻期只能一天，也就是现在是冷冻期，前一天就不可能是冷冻期

 

2）初始化：

dp[0][2]的初始化也是0

dp[0][0]=0;
dp[0][1]=-prices[0];
dp[0][2]=0;

 

3）核心代码：

i从1开始，因为0已经初始化过了

//不在冰冻期：前一个不在冰冻期/前一个就在冰冻期，现在这个变为不冰冻期
dp[i][0]=Math.max(dp[i-1][0],dp[i-1][2]);
//持股：前一个就持股/前一个不持股并且不在冰冻期，现在买入了股票
dp[i][1]=Math.max(dp[i-1][1],dp[i-1][0]-prices[i]);
//不持股在冰冻期：前一个持股卖掉了。不可能前一个还是冰冻期，冰冻期只有1天
dp[i][2]=dp[i-1][1]+prices[i];

 

4）最后返回0和2中较大的值