128 字 少于 1 分钟阅读

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

示例 1:

img

输入:n = 3
输出:5

示例 2:

输入:n = 1
输出:1

提示:

  • 1 <= n <= 19

解法

从左往右以每个节点为根节点的二叉搜索树的种类的累加。

96-2

递推公式为:dp[i] += dp[j - 1] * dp[i - j]

class Solution {
    public int numTrees(int n) {
        int[] dp = new int[n + 1];
        // 初始化0个节点和1个节点的情况
        dp[0] = 1;
        dp[1] = 1;
        for (int i = 2; i <= n; i++) {
            for (int j = 1; j <= i / 2; j++) {
                // 左子树的节点个数为j-1,右子树的节点个数为i-j
                dp[i] += dp[j - 1] * dp[i - j];
            }
            // 前(i/2)和后(i/2)数量相同,直接乘以2
            dp[i] = dp[i] * 2;
            // 若为奇数再加上根节点在中间的情况
            if (i % 2 == 1) {
                dp[i] += dp[i / 2] * dp[i / 2];
            }
        }
        return dp[n];
    }
}