和上一题的区别是这次最多只能交易两次。我的解法是给dp数组多加一列,用来标识当前状态交易次数。详情看注释,另外这里的话应该也可以进行空间优化,但是dp数组的话还是让人更好理解。
class Solution {
public int maxProfit(int[] prices) {
int len = prices.length;
//第二列的0代表当前交易完不剩股票,1代表当前交易完剩有股票
//第三列的n代表交易过的次数
int[][][] dp = new int[len][2][3];
//刚开始时第三列只能为0,第0天的股票可以选择买或者不买
dp[0][0][0] = 0;
dp[0][1][0] = -1 * prices[0];
for (int i = 1; i < len; i++) {
//1. 交易0次但手上有股票:要么是在这次买的,要么就是之前就持有
dp[i][1][0] = Math.max(dp[i - 1][0][0] - prices[i], dp[i - 1][1][0]);
//2. 交易0次且不持有股票,只有一种情况,没买过股票
dp[i][0][0] = dp[i - 1][0][0];
//3. 交易过1次手上不持有股票,涉及到是这次交易的还是上次交易的,如果是上次交易的,那么当前i值至少为2,所以根据i的取值分情况讨论
if (i >= 2)
dp[i][0][1] = Math.max(dp[i - 1][1][0] + prices[i], dp[i - 1][0][1]);
else
dp[i][0][1] = dp[i - 1][1][0] + prices[i];
//4. 交易过一次手上持有股票,同样的涉及是上次交易的还是这次交易的,另外还要设计手上持有的股票是这次就买了还是上次买的
//dp[i][1][1] = dp[i - 1][1][0];其实这里有点违和,这个的意思是我在i-1的时候卖了股票,又买了i-1的股票,有种做无用功的感觉,但是这确实是可行的一种情况,我觉得还是要考虑
if (i >= 2)
dp[i][1][1] = Math.max(dp[i - 1][1][0], Math.max(dp[i - 1][1][1], dp[i - 1][0][1] - prices[i]));
else
dp[i][1][1] = dp[i - 1][1][0];
//5. 交易过2次,手上也就无法持有股票了,涉及情况是第二次的交易是在这次还是在之前进行的,如果是在之前就交易完2次的话,i的值至少为3,因为至少要到i到2的时候才能交换两次
if (i >= 3)
dp[i][0][2] = Math.max(dp[i - 1][0][2], dp[i - 1][1][1] + prices[i]);
else if (i >= 2)
dp[i][0][2] = dp[i - 1][1][1] + prices[i];
}
//因为可能出现交换不满两次的情况,所以dp[len-1][0][1]也要考虑进来
int res = Math.max(dp[len - 1][0][2], dp[len - 1][0][1]);
return res >= 0 ? res : 0;
}
}