代码随想录算法训练营第四十一天 | 343. 整数拆分 、96.不同的二叉搜索树

学习目标

• 学习DP相关题目

---

学习内容

343. 整数拆分(Middle)

题目链接:343. 整数拆分

题目：给定一个正整数 n ，将其拆分为 k 个 正整数 的和（ k >= 2 ），并使这些整数的乘积最大化。
返回 你可以获得的最大乘积 。

示例 1:
输入: n = 2
输出: 1
解释: 2 = 1 + 1, 1 × 1 = 1。

示例 2:
输入: n = 10
输出: 36
解释: 10 = 3 + 3 + 4, 3 × 3 × 4 = 36。

提示:
2 <= n <= 58

思路：

• 确定dp数组（dp table）以及下标的含义
  dp[i] 代表的是第 i 个数拆分的乘积的最大值

• 确定递推公式
  dp[i] = Math.max(dp[i], Math.max(j*(i-j), j*dp[i-j]));
  i = 3~n, j = 1~(i-j)

• dp数组如何初始化
  dp[2] = 1

• 确定遍历顺序
  i从3~n, j 代表的是拆分, 从1~i-j, 其实就是1~i/2一直找拆分乘积, 取最大的

• 举例推导dp数组

时间复杂度：O(n^2) 空间复杂度：O(n)

解决方案：

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

小结：已经开始不大好找dp[i]及其对应下标意义了

96.不同的二叉搜索树(Middle)

题目链接:96.不同的二叉搜索树

题目：给你一个整数 n ，求恰由 n 个节点组成且节点值从 1 到 n 互不相同的 二叉搜索树 有多少种？返回满足题意的二叉搜索树的种数。

示例 1：
输入：n = 3
输出：5

示例 2：
输入：n = 1
输出：1

提示：
1 <= n <= 19

思路：
这里引入 Carl 哥的分析:

dp[3]，就是 元素1为头结点搜索树的数量 + 元素2为头结点搜索树的数量 + 元素3为头结点搜索树的数量
元素1为头结点搜索树的数量 = 右子树有2个元素的搜索树数量 * 左子树有0个元素的搜索树数量
元素2为头结点搜索树的数量 = 右子树有1个元素的搜索树数量 * 左子树有1个元素的搜索树数量
元素3为头结点搜索树的数量 = 右子树有0个元素的搜索树数量 * 左子树有2个元素的搜索树数量
有2个元素的搜索树数量就是dp[2]。
有1个元素的搜索树数量就是dp[1]。
有0个元素的搜索树数量就是dp[0]。
所以dp[3] = dp[2] * dp[0] + dp[1] * dp[1] + dp[0] * dp[2]

• 确定dp数组（dp table）以及下标的含义
  dp[i], 就是i 个节点组成且节点值从 1 到 i 互不相同的 二叉搜索树数量, 根据上面的分析也是元素头结点为 1~i 的有i个节点的搜索树的数量.
• 确定递推公式
  dp[i] += dp[j-1] * dp[i-j], i 从2~n, j 从 1 ~ i.
• dp数组如何初始化
  dp[0] = 0, dp[1]=1
• 确定遍历顺序
  i 从2~n, j 从 1 ~ i.
• 举例推导dp数组

时间复杂度：O(n^2) 空间复杂度：O(n)

解决方案：

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

小结：

---

总结

感觉这是一类 dp 题目的思考方式, 虽然仍然是 5 步思考, 但dp数组（dp table）以及下标的含义与昨天不同