最多可以完成 两笔 交易
输入: [3,3,5,0,0,3,1,4]
输出: 6
解释: 在第 4 天(股票价格 = 0)的时候买入,在第 6 天(股票价格 = 3)的时候卖出,这笔交易所能获得利润 = 3-0 = 3 。
随后,在第 7 天(股票价格 = 1)的时候买入,在第 8 天 (股票价格 = 4)的时候卖出,这笔交易所能获得利润 = 4-1 = 3
public int maxProfit(int[] prices) {
//三维DP动态规划。三个维度,第一维表示天,第二维表示交易了几次,第三维表示是否持有股票
if(prices == null) return 0;
int len = prices.length;
if(len == 0) return 0;
int k = 2;
//dp[i][j][0] 今天是第i天 进行 j次 交易 手上没有股票
//dp[i][j][1] 今天是第i天 进行 j次 交易 手上有股票
int[][][] dp = new int[len][k + 1][2];
for(int i = 0; i < len; i ++) {
for(int j = k; j > 0; j --) {
//base case
if(i == 0) {
dp[0][j][0] = 0;
dp[0][j][1] =- prices[0];
continue;
}
今天手里没股票,比较MAX(前一天可能没股票,前一天有股票但是今天卖出去了,卖出去就有利润,所以+ prices[i])
dp[i][j][0] = Math.max(dp[i - 1][j][0], dp[i - 1][j][1] + prices[i]);
今天手里有股票,比较MAX(前一天可能有股票,前一天没股票但是今天买了,买了就有成本,所以- prices[i])
dp[i][j][1] = Math.max(dp[i - 1][j][1], dp[i - 1][j - 1][0] - prices[i]);
}
}
//肯定是手里没股票了 利润大
return dp[len - 1][k][0];
}