题干:
代码:
class Solution {
public:
int maxProfit(vector<int>& prices) {
if (prices.size() < 2) return 0;
int buy1 = prices[0];
int buy2 = prices[0];
int sell1 = 0, sell2 = 0;
for (int i = 1; i < prices.size(); ++i) {
buy1 = min(buy1, prices[i]);
sell1 = max(sell1, prices[i] - buy1);
buy2 = min(buy2, prices[i] - sell1);
sell2 = max(sell2, prices[i] - buy2);
}
return sell2;
}
};
buy1
和buy2
分别存储了第一笔交易的买入价格和第二笔交易的买入价格(即卖出第一笔股票后的价格)。sell1
和sell2
则分别代表了第一笔交易和第二笔交易的卖出价格。
我们遍历价格数组,不断更新这四个值:如果当前价格小于buy1
,那么更新buy1
;如果当前价格减去buy1
大于sell1
,那么更新sell1(第一次卖出最大利润)
;更新buy2为当前价格减去sell1和之前的buy2中的较小值,表示在当前价格下第二次买入股票时的最低花费, 通过不断更新 buy2
,我们可以确保在第二次买入时不会支付比第一次卖出后更高的价格,从而避免亏损;如果当前价格减去buy2
大于sell2
,那么更新sell2
。
最后,sell2
就是我们想要的结果:两次交易所能获得的最大利润。