Say you have an array for which the ith element is the price of a given stock on day i.
Design an algorithm to find the maximum profit. You may complete at most two transactions.
Note:
You may not engage in multiple transactions at the same time (ie, you must sell the stock before you buy again).
题意
股票交易问题,只不过最多可以买卖两次,同时必须等卖完第一次的才可以买第二次的。
思路:
与121. Best Time to Buy and Sell Stock的第二种解法类似,设置4种情况,找出他们之间的联系,写出动态资源方程即可,同时也要注意初始化问题,并且因为可能只交易一次,最后结果取交易两次和交易一次的最大值。
代码:
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n=prices.size();
int profit[4]={INT_MIN,0,INT_MIN,0};
for(int i=0;i<n;i++)
{
profit[3]=max(profit[3],profit[2]+prices[i]);
profit[2]=max(profit[2],profit[1]-prices[i]);
profit[1]=max(profit[1],profit[0]+prices[i]);
profit[0]=max(profit[0],-prices[i]);
}
return max(profit[1],profit[3]);
}
};