96. 不同的二叉搜索树

96. 不同的二叉搜索树

题目描述

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

示例1：

输入：n = 3
输出：5

示例2：

输入：n = 1
输出：1

提示：

1 <= n <= 19

题解

思路

• dp[i]表示1->i能表示的不同二叉搜索树的个数
• 对于数字i，考虑1->i分别去当二叉树根的情况
• 由于二叉搜索树的定义规则，我们可以得到
  • 对于以i为根节点的二叉搜索树(共有n个节点)来说
  • 其左子树一定有i-1个节点
  • 其右子树一定有n-j个节点
• 所以根据分布乘法原理
• 有递推式 dp[i] = dp[j-1] * dp[i-j] 其中j就是选择小于等于i的所有节点都去充当头节点的可能

代码

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];
    }
}