动态规划之股票问题

动态规划之股票问题

1. “121. 买卖股票的最佳时机”

](https://i0.hdslb.com/bfs/album/a10d1c5b8ea00e9379a0b2b8439d2330b62962a6.png)
首先确定dp数组的含义，dp[i][0]代表第i天不持有股票所剩余金额，dp[i][1]代表第i天持有股票所剩余金额。
那么第i天不持有股票可以是前一天就不持有股票，即dp[i-1][0],也可以是前一天持有股票，第i天卖出股票即dp[i-1][1]+prices[i]，两者取最大值，列出状态转移方程dp[i][0]=Math.max(dp[i-1][0],dp[i-1][1]+prices[i])。
同样的，第i天持有股票可以是前一天就持有股票，即dp[i-1][1],也可以是前一天不持有股票，第i天买入股票即dp[i-1][0]-prices[i],两者取最大值，列出状态转移方程dp[i][1]=Math.max(dp[i-1][1],dp[i-1][0]-prices[i])。
因此得出
dp[i][0]=Math.max(dp[i-1][0],dp[i-1][1]+prices[i])
dp[i][1]=Math.max(dp[i-1][1],dp[i-1][0]-prices[i])
然后因为本题因为只能买入一次，买入股票后所剩余金额必定等于0-股票价格，因此dp[i][1]简化为dp[i][1]=Math.max(dp[i-1][1],-prices[i])
最后，因为取值只依赖于dp[0]和dp[1]，简化为一维数组

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

2. “122. 买卖股票的最佳时机 II”

整体思路与上一题一致，不过不同的是这题可以无限次的买卖同一只股票，这就意味着你买入股票剩余的金额不再是-prices[i]了，而应该是前一天不持有股票所剩金额-prices[i]
即dp[i][1]=Math.max(dp[i-1][1],dp[i-1][0]-prices[i])
而di[i][0]第i天不持有股票和上一题一致dp[i][0]=Math.max(dp[i-1][0],dp[i-1][1]+prices[i])

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

3. “123. 买卖股票的最佳时机III”

这一题与上两题不同的是交易次数限制为两次，因此设置一个未知数k代表交易次数，并且dp数组可以些为dp[i][k][0]和dp[i][k][1]代表最多能有k次交易
并且一次交易代表着买入和卖出两个操作，因此我们只需在买入时将k-1就可以了，所以状态转移方程为
dp[i][k][0] = Math.max(dp[i - 1][k][0], dp[i - 1][k][1] + prices[i])
dp[i][k][1] = Math.max(dp[i - 1][k][1], dp[i - 1][k - 1][0] - prices[i])
本题交易次数限制为两次，将k=2带入得到

dp[i][2][0] = Math.max(dp[i - 1][2][0], dp[i - 1][2][1] + prices[i])  
dp[i][2][1] = Math.max(dp[i - 1][2][1], dp[i - 1][1][0] - prices[i])  
dp[i][1][0] = Math.max(dp[i - 1][1][0], dp[i - 1][1][1] + prices[i])  
dp[i][1][1] = Math.max(dp[i - 1][1][1], dp[i - 1][0][0] - prices[i])  
= Math.max(dp[i - 1][1][1], -prices[i])

简化一下就是

dp_i20 = Math.max(dp_i20, dp_i21 + prices[i])
dp_i21 = Math.max(dp_i21, dp_i10 - prices[i])
dp_i10 = Math.max(dp_i10, dp_i11 + prices[i])
dp_i11 = Math.max(dp_i11, -prices[i])

然后考虑他们的初值，第0天不持有，剩余金额肯定为0，因此dp_i20和dp_i10初值为0
第0天持有，剩余金额肯定为-prices[0]，因此dp_i21和dpi11初值为-prices[0]

class Solution {
    public int maxProfit(int[] prices) {
        int i02 = 0;
        int i12 = 0 - prices[0];
        int i01 = 0;
        int i11 = 0 - prices[0];
        for (int i = 0; i < prices.length; i++) {
            i02 = Math.max(i02, i12 + prices[i]);
            i12 = Math.max(i12, i01 - prices[i]);
            i01 = Math.max(i01, i11 + prices[i]);
            i11 = Math.max(i11, -prices[i]);
        }
        return i02;
    }
}

4. “188. 买卖股票的最佳时机 IV”

这题的条件改为了交易次数最多为k次，求最大利润。思路和上一题是一模一样的，只不过上一次k=2可以把情况穷举出来，而这题k是未知数，因此使用三维数组。

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

不过要注意交易次数最大为prices[]的长度除以2才是有效的，因为一次买入必须对应着一次卖出，交易次数最多为prices[]的长度除以2。

5. “309. 最佳买卖股票时机含冷冻期”

这一题的条件为卖出股票后无法在第二天买入股票，也就是说在第i天卖出股票，最少得在第i+2天才能买入，
也就是说dp[i][1]应该为dp[i][1]=Math.max(dp[i-1][1],dp[i-2][0]-prices[i])，但是当i=1时，也就是在第1天持有股票时，股票肯定是在第0天或者第1天买入的
然后并没有规定买入股票有冷冻期，因此买入股票可以在第二天卖出，dp[i][0]不变还是dp[i][0]=Math.max(dp[i-1][0],dp[i-1][1]+prices[i])

class Solution {
    public int maxProfit(int[] prices) {
        int[][] dp = new int[prices.length][2];
        dp[0][0] = 0;
        dp[0][1] = -prices[0];
        for (int i = 1; i < prices.length; i++) {
            dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][1] + prices[i]);
            dp[i][1] = i > 1 ? Math.max(dp[i - 1][1], dp[i - 2][0] - prices[i]) : Math.max(dp[i - 1][1], dp[i - 1][0] - prices[i]);
        }
        return dp[prices.length - 1][0];
    }
}

6. “714. 买卖股票的最佳时机含手续费”

这题的条件加了一个手续费，每笔交易都需支付一个固定的手续费，我们只需在卖出时减去手续费就行了，还是熟悉的套路

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