Merge pull request #480 from hk27xing/hk27xing-add

改进123, 188的 Java 版本代码

problems/0188.买卖股票的最佳时机IV.md

@@ -170,41 +170,54 @@ public:
 Java：
 
 ```java
-class Solution { //动态规划
+// 版本一: 三维 dp数组
+class Solution {
     public int maxProfit(int k, int[] prices) {
-        if (prices == null || prices.length < 2 || k == 0) {
-            return 0;
+        if (prices.length == 0) return 0;
+
+        // [天数][交易次数][是否持有股票]
+        int len = prices.length;
+        int[][][] dp = new int[len][k + 1][2];
+        
+        // dp数组初始化
+        // 初始化所有的交易次数是为确保 最后结果是最多 k 次买卖的最大利润
+        for (int i = 0; i <= k; i++) {
+            dp[0][i][1] = -prices[0];
         }
 
-        // [天数][交易次数][是否持有股票] 
-        int[][][] dp = new int[prices.length][k + 1][2];
-
-        // bad case
-        dp[0][0][0] = 0;
-        dp[0][0][1] = Integer.MIN_VALUE;
-        dp[0][1][0] = 0;
-        dp[0][1][1] = -prices[0];
-        // dp[0][j][0] 都均为0
-        // dp[0][j][1] 异常值都取Integer.MIN_VALUE;
-        for (int i = 2; i < k + 1; i++) {
-            dp[0][i][0] = 0;
-            dp[0][i][1] = Integer.MIN_VALUE;
-        }
-
-        for (int i = 1; i < prices.length; i++) {
-            for (int j = k; j >= 1; j--) {
-                // dp公式
+        for (int i = 1; i < len; i++) {
+            for (int j = 1; j <= k; j++) {
+                // dp方程, 0表示不持有/卖出, 1表示持有/买入
                 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[len - 1][k][0];
+    }
+}
 
-        int res = 0;
-        for (int i = 1; i < k + 1; i++) {
-            res = Math.max(res, dp[prices.length - 1][i][0]);
+// 版本二: 空间优化
+class Solution {
+    public int maxProfit(int k, int[] prices) {
+        if (prices.length == 0) return 0;
+
+        // [天数][股票状态]
+        // 股票状态: 奇数表示第 k 次交易持有/买入, 偶数表示第 k 次交易不持有/卖出, 0 表示没有操作
+        int len = prices.length;
+        int[][] dp = new int[len][k*2 + 1];
+        
+        // dp数组的初始化, 与版本一同理
+        for (int i = 1; i < k*2; i += 2) {
+            dp[0][i] = -prices[0];
         }
 
-        return res;
+        for (int i = 1; i < len; i++) {
+            for (int j = 0; j < k*2 - 1; j += 2) {
+                dp[i][j + 1] = Math.max(dp[i - 1][j + 1], dp[i - 1][j] - prices[i]);
+                dp[i][j + 2] = Math.max(dp[i - 1][j + 2], dp[i - 1][j + 1] + prices[i]);
+            }
+        }
+        return dp[len - 1][k*2];
     }
 }
 ```