题干:
代码:
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
就是我们想要的结果:两次交易所能获得的最大利润。