最多进行两次买卖股票,并且不能再卖出前继续买,所以两次交易必须是分布在2个区间内
我们可以划分2个区间为[0,i]和[i,prices.length-1],注意这样就把原题转换成了和Best time to Buy and Sell Stock I 一样了,求两个区间上各自的最大值,最后加起来就是最大利润。那么为什么左和右区间可以有overlap的位置呢?因为没有规定不能在同一天卖完再买,所以这个重合的位置就代表着我先卖掉再立刻买进。
下面是O(N*N) 的方法。TLE无法通过大集合,所以迫使我认怂去网上找答案。
public int maxProfit(int[] prices) {
if(prices==null || prices.length < 2)
return 0;
int i = 0;
int maxR = 0;
int maxL = 0;
int minV = Integer.MIN_VALUE;
int maxP = 0;
while(i < prices.length) {
minV = Integer.MIN_VALUE;
int l = 0;
while(l < i) {
minV = Math.min(minV, prices[l]);
maxL = Math.max(maxL, prices[l] - minV);
l++;
}
int r = i;
while( r < prices.length) {
minV = Math.min(minV, prices[r]);
maxR = Math.max(maxR, prices[r] - minV);
r++;
}
maxP = Math.max(maxP, maxR + maxL);
i++;
}
return maxP;
}
OK, 在网上上找到了动态规划版。最简单的思路是,从左到右存储发现的最大值,然后从右到左再寻找最大值和dp里存的值相加的最大值,相当于不用重复计算已经计算过的左边的值了,所以就只是两遍遍历,时间复杂度还是O(N), 空间复杂度是O(N)。
public int maxProfit(int[] prices) {
if(prices==null || prices.length < 2)
return 0;
int i = 0;
int[] dp = new int[prices.length];
int minV = Integer.MAX_VALUE;
int maxP = 0;
int maxV = Integer.MIN_VALUE;
while(i < prices.length) {
minV = Math.min(minV, prices[i]);
maxP = Math.max(maxP, prices[i] - minV);
dp[i] = maxP;
i++;
}
int maxPBack = 0;
i = prices.length - 1;
maxP = 0;
while( i >= 0) {
maxV = Math.max(maxV, prices[i]);
maxPBack = Math.max(maxPBack, maxV - prices[i] );
maxP = Math.max(maxPBack + dp[i], maxP);
i--;
}
return maxP;
}
稍微麻烦一点的还可以三遍遍历,左边到右遍历和右边到左遍历的值都相加存起来,再找这个值的最大值。就不多写了。