123. 买卖股票的最佳时机 III（困难）

思路：

与122不同的时，当前限制了只能交易2笔，则交易多少笔，就设多少种相应状态

有5种状态：

    i）不持股，没有任何操作（0）

    ii）第一次买入（1）

    iii）第一次卖出（2）

    iiii）第二次买入（3）

    iiiii）第二次卖出（4）

 

代码：

class Solution {
    public int maxProfit(int[] prices) {
		int n=prices.length;
		if(n<2) return 0;
		
		//状态转移方程
		//没有任何操作（0）
		//第一次买入（1）
		//第一次卖出（2）
		//第二次买入（3）
		//第二次卖出（4）
		
		int[][] dp=new int[n][5];
		
		//初始化
		dp[0][0]=0;
		dp[0][1]=-prices[0];
		
		//状态3赋值为一个不可能的数，因为只有前面的所有状态都被赋值时，才能赋值3
		for(int i=0;i<n;i++){
			dp[i][3]=Integer.MIN_VALUE;
		}
		
		for(int i=1;i<n;i++){
			dp[i][0]=0;
			//第一次买入：前一次就是已买入状态/前一次没有任何操作，当前买入
			dp[i][1]=Math.max(dp[i-1][1],dp[i-1][0]-prices[i]);
			//第一次卖出：前一次就是已卖出状态/前一次是已买入状态，当前卖出
			dp[i][2]=Math.max(dp[i-1][2],dp[i-1][1]+prices[i]);
			//第二次买入：前一次就是第二次已买入状态/前一次是第一次卖出，当前买入
			dp[i][3]=Math.max(dp[i-1][3],dp[i-1][2]-prices[i]);
			//第二次卖出：前一次就是已卖出/前一次是第二次买入状态，当前卖出
			dp[i][4]=Math.max(dp[i-1][4],dp[i-1][3]+prices[i]);
		}
		//只比较已卖出的状态获利值
		return Math.max(dp[n-1][0],Math.max(dp[n-1][2],dp[n-1][4]));
    }
}

 

分解：

1）是dp的第一种形式，当前值只与前2个结果有关

区间规划

动态转移方程：

dp[i][0]=0;
//第一次买入：前一次就是已买入状态/前一次没有任何操作，当前买入
dp[i][1]=Math.max(dp[i-1][1],dp[i-1][0]-prices[i]);
//第一次卖出：前一次就是已卖出状态/前一次是已买入状态，当前卖出
dp[i][2]=Math.max(dp[i-1][2],dp[i-1][1]+prices[i]);
//第二次买入：前一次就是第二次已买入状态/前一次是第一次卖出，当前买入
dp[i][3]=Math.max(dp[i-1][3],dp[i-1][2]-prices[i]);
//第二次卖出：前一次就是已卖出/前一次是第二次买入状态，当前卖出
dp[i][4]=Math.max(dp[i-1][4],dp[i-1][3]+prices[i]);

 

2)状态3赋值为一个不可能的数，因为只有前面的所有状态都被赋值时，才能赋值3

for(int i=0;i<n;i++){
    dp[i][3]=Integer.MIN_VALUE;
}

 

3）最后要比较卖出的结果，则比较状态0，2，4

而Math.max()只能有2个参数，则在其中再嵌套一个Math.max()

return Math.max(dp[n-1][0],Math.max(dp[n-1][2],dp[n-1][4]));