[leetcode] 123. Best Time to Buy and Sell Stock III

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题的基础上将操作次数限定为两次，难度为Hard。

每一天的操作分三种情况，不操作、买入或卖出，不操作的情况最大收益不变。初始收益为0，买入时收益减买入价格，卖出时收益加卖出价格。在处理121题时我们将每天的状态分为两种，买入之后没有卖出和卖出两种，这道题允许操作两次，因而每天状态变更如下：

• profit[0]：买入第一次，没有卖出

• profit[1]：卖出第一次，没有买入

• profit[2]：买入第二次，没有卖出

• profit[3]：卖出第二次

参考第121题我们就能够推出新的状态机状态迁移策略，具体代码：

class Solution {
public:
    int maxProfit(vector<int>& prices) {
        int profit[4] = {INT_MIN, 0, INT_MIN, 0};
        
        for(int i=0; i<prices.size(); 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]);
    }
};

最后取profit[1]和profit[3]的最大值是因为可能操作一次取得最大收益。