DP有點難。
看到一個不錯的講解,對應的轉移方程:
- 在持有 buy[i] = max(sell[i-2] - price[i], buy[i-1])
- 不在持有 sell[i] = max(buy[i-1] + price[i],sell[i-1])
代碼的話像覃超說的,轉移方程出來了就不要動腦子了。注意Java這邊不太好用-1做數(shù)組下標。
public int maxProfit(int[] prices) {
if (prices.length < 2) return 0;
int[] buy = new int[prices.length];
int[] sell = new int[prices.length];
buy[0] = -prices[0];
buy[1] = Math.max(-prices[0] , -prices[1]);
sell[0] = 0;
sell[1] = Math.max(prices[1] - prices[0] , 0 );
for (int i = 2; i < prices.length; i++) {
sell[i] = Math.max(buy[i - 1] + prices[i], sell[i - 1]);
buy[i] = Math.max(sell[i - 2] - prices[i], buy[i - 1]);
}
return sell[sell.length - 1];
}
}