思路
本题是最多有两次买进股票的机会,与121题很类似,只是多了一次买卖的机会。还可以套用121题的思路
LeetCode121. 买卖股票的最佳时机_想进阿里的小菜鸡的博客-CSDN博客
就是多了两种状态。其递推公式如下
dp[i][0]=Math.max(dp[i-1][0],-prices[i]);
dp[i][1]=Math.max(dp[i-1][0]+prices[i],dp[i-1][1]);
dp[i][2]=Math.max(dp[i-1][2],dp[i-1][1]-prices[i]);
dp[i][3]=Math.max(dp[i-1][2]+prices[i],dp[i-1][3]);
代码
class Solution {
public int maxProfit(int[] prices) {
int dp[][] = new int[prices.length][4];
dp[0][0] = -prices[0];
dp[0][1]=0;
dp[0][2]=-1000000;
dp[0][3]=-1000000;
for(int i =1;i<prices.length;i++){
dp[i][0]=Math.max(dp[i-1][0],-prices[i]);
dp[i][1]=Math.max(dp[i-1][0]+prices[i],dp[i-1][1]);
dp[i][2]=Math.max(dp[i-1][2],dp[i-1][1]-prices[i]);
dp[i][3]=Math.max(dp[i-1][2]+prices[i],dp[i-1][3]);
}
return Math.max(Math.max(dp[prices.length-1][0],dp[prices.length-1][1]),Math.max(dp[prices.length-1][2],dp[prices.length-1][3]));
}
}