给你一个整数 n ,求恰由 n 个节点组成且节点值从 1 到 n 互不相同的 二叉搜索树 有多少种?返回满足题意的二叉搜索树的种数。
示例 1:
输入:n = 3
输出:5
示例 2:
输入:n = 1
输出:1
提示:
1 <= n <= 19
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/unique-binary-search-trees
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution {
Map<Integer, Integer> map = new HashMap<>();
public int numTrees(int n) {
// return numTreesI(n);
// return numTreesII(n);
return numTreesIII(n);
}
//方法三:动态规划
//时间复杂度O(N^2),空间复杂度O(N)
private int numTreesIII(int n) {
int[] dp = new int[n + 1];
dp[0] = 1;
dp[1] = 1;
//第一层循环是遍历n,第二层循环表示以i为根节点的BST个数
//例如n=5,表示{1,2,3,4,5}组成BST的个数,也就等于分别以1~5为根节点组成BST个数的总数
for (int i = 2; i <= n; i++) {
for (int j = 1; j <= i; j++) {
//这里表示以i为根节点的BST个数等于左子树个数 * 右子树个数
dp[i] += dp[j - 1] * dp[i - j];
}
}
return dp[n];
}
//方法二:方法一基础上增加缓存
//时间复杂度O(N^2),空间复杂度O(N)
private int numTreesII(int n) {
if (n == 0 || n == 1) {
return 1;
}
if (map.containsKey(n)) {
return map.get(n);
}
int res = 0;
for (int i = 1; i <= n; i++) {
int left = numTreesII(i - 1);
int right = numTreesII(n - i);
res += left * right;
}
map.put(n, res);
return res;
}
//方法一:递归
//以i为根节点的BST个数等于左子树的组合数 * 右子树的组合数
//Time Limit Exceeded
private int numTreesI(int n) {
if (n == 0 || n == 1) {
return 1;
}
int res = 0;
for (int i = 1; i <= n; i++) {
int left = numTreesI(i - 1);
int right = numTreesI(n - i);
res += left * right;
}
return res;
}
}