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

思路：

动态规划dp

 

代码：

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）此题限制2次购买，所以状态很多，要覆盖2次购买的买入/卖出状态

初始化：

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

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

 

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]);

返回已卖出的状态获利值最大的一个

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

 

复杂度分析：

时间复杂度：O（N）

空间复杂度：O（N）