每天的操作有五种状态:
- 不操作
- 第一次买入
- 第一次卖出
- 第二次买入
- 第二次卖出
不操作无需更新状态
每天更新其余四个买卖状态的最大收益(第一次买入、第一次卖出、第二次买入、第二次卖出)
由于求最大收益,所以初始状态为第一天买入不卖出(既最小收益)。
最后返回完成两次交易的最大收益。
class Solution {
public:
int maxProfit(vector<int>& prices) {
int n=prices.size();
int buy1=-prices[0],sell1=0;
int buy2=-prices[0],sell2=0;
for(int i=1;i<n;i++){
buy1=max(-prices[i],buy1);
sell1=max(sell1,buy1+prices[i]);
buy2=max(buy2,sell1-prices[i]);
sell2=max(sell2,buy2+prices[i]);
}
return sell2;
}
};