给你一个整数 n
,求恰由 n
个节点组成且节点值从 1
到 n
互不相同的 二叉搜索树 有多少种?返回满足题意的二叉搜索树的种数。
输入范围:1 <= n <= 19
思路分析:二叉搜索树的特点:每一个节点的左子树上的所有元素大小都小于该节点,右子树上的所有元素大小都大于该节点。按照题意,如果有n个从小到大排列的节点,那么所有的二叉搜索树的个数之和其实就等于,每一个节点做根节点的二叉搜索树之和。而每一个根节点都可以将问题分解为左子树之和*右子树之和。所以我们可以有两种做法,做法一:动态规划打表,做法二:抖机灵。其实还有一个数学方法,奈何我是渣渣看不懂。
方法一:动态规划
class Solution {
public int numTrees(int n) {
int[] nums = new int[n+1];
nums[0] = nums[1] = 1;
for(int i = 2 ; i <= n ; i++){
for(int l = 0 ; l <= i-1 ; l++){
int r = i - l - 1;
nums[i] += nums[l]*nums[r];
}
}
return nums[n];
}
}
其中nums[]数组每个元素存放的都是有n个节点情况下二叉搜索树的总和,数组每个位置的计算公式便是nums[i] += nums[l]*nums[r];将每个节点作为根节点的二叉搜索树之和。(特别注意,当根节点为1或者n时,对应的l和r分别是0,所以我们初始化之前,让nums[0] = 1)
方法二:抖机灵(面试会被打死)
我们可以发现题目数据输入发现最大为19,所以我们可以把每种情况都算出来,然后return。、
class Solution {
public int numTrees(int n) {
switch(n){
case 1: return 1;
case 2: return 2;
case 3: return 5;
case 4: return 14;
case 5: return 42;
case 6: return 132;
case 7: return 429;
case 8: return 1430;
case 9: return 4862;
case 10: return 16796;
case 11: return 58786;
case 12: return 208012;
case 13: return 742900;
case 14: return 2674440;
case 15: return 9694845;
case 16: return 35357670;
case 17: return 129644790;
case 18: return 477638700;
case 19: return 1767263190;
default: return 0;
}
}
}
方法三:补充数学方法:(卡塔兰数)
class Solution {
public int numTrees(int n) {
// 提示:我们在这里需要用 long 类型防止计算过程中的溢出
long C = 1;
for (int i = 0; i < n; ++i) {
C = C * 2 * (2 * i + 1) / (i + 2);
}
return (int) C;
}
}