问题
给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。
注意: 你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
例子
思路
可化为两个子问题:
收益[0:n] = max(一次买卖收益[0:i]+一次买卖收益[i:n]) 其中i 为 0->n;
代码
//粗糙
class Solution {
public int maxProfit(int[] nums) {
int res = 0;
for(int i=0; i<nums.length; i++) {
int a = get(nums, 0, i);
int b = get(nums, i, nums.length-1);
res = Math.max(a+b, res);
}
return res;
}
public int get(int[] nums, int low, int high) {
//buy:到i之前最小的数
int buy=nums[low],res=0;
low++;
for(int i=low; i<=high; i++) {
res = Math.max(res, nums[i]-buy);
buy = Math.min(buy,nums[i]);
}
return res;
}
}
//升级
class Solution {
public int maxProfit(int[] nums) {
if(nums.length==0) return 0;
int[] dp1=new int[nums.length],dp2=new int[nums.length];
//buy:0到i-1最小的数,sell:len-1~i+1最大的数
int buy=nums[0],sell=nums[nums.length-1];
for(int i=1; i<nums.length; i++) {
dp1[i]=Math.max(dp1[i-1], nums[i]-buy);
buy=Math.min(buy,nums[i]);
}
for(int i=nums.length-2; i>=0; i--) {
dp2[i]=Math.max(dp2[i+1],sell-nums[i]);
sell=Math.max(sell,nums[i]);
}
int res=0;
for(int i=0; i<nums.length; i++) {
res = Math.max(res,dp1[i]+dp2[i]);
}
return res;
}
}