LeetCode 123 买卖股票的最佳时机 III 题解

题目大意

与前两题类似，不过这里要求只进行两次交易，然后求解最大利润。

分析

不妨设dp[i][j][k]表示第i天，至多j笔交易，持有股票(k=1)或不持有股票(k=0)时，能够获取到的最大利润。那么很显然，有如下结论：

如果第j天完成至多j笔交易的话，那么它要么保持前一天的状态不变，即不发生交易，要么在今天买入股票。

同理，如果第j天完成至多j笔交易且不持有股票的话，那么它要么保持前一天的状态不变，即不发生交易，要么在今天卖出股票。

有了这个状态转移方程，可以很容易写出程序：

public int maxProfit(int[] prices) {
        if(prices.length == 0) {
            return 0;
        }
        int dp[][][] = new int[prices.length][3][2];
        for(int i = 1; i < 3; i++) {
            dp[0][i][1] = -prices[0];
        }
        for(int i = 1; i < prices.length; i++) {
            for(int j = 1; j <= 2; j++) {
                dp[i][j][0] = Math.max(dp[i - 1][j][0], dp[i - 1][j][1] + prices[i]);
                dp[i][j][1] = Math.max(dp[i - 1][j][1], dp[i - 1][j - 1][0] - prices[i]);
            }       
        }
        return dp[prices.length - 1][2][0];
    }