算法训练营第四十三天 | LeetCode 343 整数拆分、LeetCode 96 不同的二叉搜索树

LeetCode 343 整数拆分

---

这题其实可以用数学规律:乘积最大时总是分成的数最均衡时取到。

代码如下：

class Solution {
    public int integerBreak(int n) {
        int[] dp = new int[n + 1];
        dp[0] = 0;
        dp[1] = 0;
        int max = 0;
        for (int i = 2; i <= n; i++) {
            if (n % i == 0) {
                dp[i] = (int)Math.pow(n / i, i);
            } else {
                dp[i] = (int)(Math.pow(n / i + 1, n % i)*Math.pow(n / i, i - n % i));
            }
            if (max < dp[i]) max = dp[i];
        }
        return max;
    }
}

这题也可以用动规来写，dp数组含义是该数被拆分成k（k >= 2）个数的最大乘积。下标就是该数，遍历顺序也可以知道是从前往后遍历。

但这里递推逻辑要注意下，其实感觉看代码也能看得懂。递推逻辑中加入两个Math.max是要比较该数拆分后和原数之间哪个大，取比较大的那个。

代码如下：

class Solution {
    public int integerBreak(int n) {
        int[] dp = new int[n + 1];
        dp[0] = 0;
        dp[1] = 1;
        for (int i = 2; i <= n; i++) dp[i] = 0;
        for (int i = 2; i <= n; i++) {
            for (int j = 1; j <= i / 2; j++) {
                dp[i] = Math.max(dp[i], Math.max(j, dp[j])*Math.max(i - j, dp[i - j]));
            }
        }
        return dp[n];
    }
}

LeetCode 96 不同的二叉搜索树

class Solution {
    public int numTrees(int n) {
        if (n <= 2) return n;
        int[] dp = new int[n + 1];
        dp[0] = 1;
        dp[1] = 1;
        dp[2] = 2;
        for (int i = 3; i <= n; i++) {
            for (int j = 1; j <= i; j++) {
                dp[i] += dp[j - 1]*dp[i - j];
            }
        }
        return dp[n];
    }
}