【Leetcode】123. Best Time to Buy and Sell Stock III

题目地址：

https://leetcode.com/problems/best-time-to-buy-and-sell-stock-iii/

给定一个股票的价格，以长$n$的数组$p$的形式给出。规定每天只能做一次交易，并且每次只能买$1$股，问如果只允许做最多$2$次买卖，那么最大收益是多少。

思路是前后缀分解。首先预处理出给定前缀$p[0:k]$且只进行一次交易的情况下的最大收益，这个可以用动态规划做。设$f[k]$是上述前缀的最大收益，那么$f[0]=0$， f [ k ] = max ⁡ { f [ k − 1 ] , p [ k ] − m } , k ≥ 1 f[k]=\max\{f[k-1],p[k]-m\},k\ge 1 f[k]=max{f[k−1],p[k]−m},k≥1，$m=\min p[0:k-1]$。算出$f$之后，可以枚举第二次交易的买入点。设第二次交易的买入点是$u$，那么总的最大收益是$(M-p[u])+f[u]$，其中$M=\max p[u+1:n-1]$。这里注意，当第一次卖出点和第二次买入点重合的时候，我们实际上也同时枚举了只交易一次的情况。所以必须得写$(M-p[u])+f[u]$而不能写$(M-p[u])+f[u-1]$。代码如下：

public class Solution {
    public int maxProfit(int[] prices) {
        if (prices == null || prices.length <= 1) {
            return 0;
        }
        
        int n = prices.length;
        int[] dp = new int[n];
        for (int i = 1, min = prices[0]; i < n; i++) {
            dp[i] = Math.max(dp[i - 1], prices[i] - min);
            min = Math.min(min, prices[i]);
        }
        
        int res = 0;
        for (int i = n - 1, max = 0; i >= 0; i--) {
            res = Math.max(res, max - prices[i] + dp[i]);
            max = Math.max(max, prices[i]);
        }
        
        return res;
    }
}

时空复杂度$O(n)$。