我的思路是前i天的最大收入加上后n-i天的最大收入之和,即为最大收入,但代码执行超时。
class Solution {
public int maxProfit(int[] prices) {
if (prices.length<=1) return 0;
int max_1 = 0;
int max_2 = 0;
int max = 0;
for (int i = 0; i < prices.length; i++) {
int min = prices[0];
for (int j = 0; j < i; j++) {
max_1 = Math.max(prices[j]-min,max_1);
min = Math.min(prices[j], min);
}
min = prices[i];
for (int j = i; j < prices.length; j++) {
max_2 = Math.max(prices[j]-min,max_2);
min = Math.min(prices[j], min);
}
if (max_1 + max_2 > max) max = max_1 + max_2;
max_1 = max_2 = 0;
}
return max;
}
}
评论区大神答案,动态规划
class Solution {
public int maxProfit(int[] prices) {
/**
对于任意一天考虑四个变量:
fstBuy: 在该天第一次买入股票可获得的最大收益
fstSell: 在该天第一次卖出股票可获得的最大收益
secBuy: 在该天第二次买入股票可获得的最大收益
secSell: 在该天第二次卖出股票可获得的最大收益
分别对四个变量进行相应的更新, 最后secSell就是最大
收益值(secSell >= fstSell)
**/
int fstBuy = Integer.MIN_VALUE, fstSell = 0;
int secBuy = Integer.MIN_VALUE, secSell = 0;
for(int p : prices) {
fstBuy = Math.max(fstBuy, -p);
fstSell = Math.max(fstSell, fstBuy + p);
secBuy = Math.max(secBuy, fstSell - p);
secSell = Math.max(secSell, secBuy + p);
}
return secSell;
}
}