给定一个整数 n,求以 1 ... n 为节点组成的二叉搜索树有多少种?
示例:
输入: 3
输出: 5
解释:
给定 n = 3, 一共有 5 种不同结构的二叉搜索树:
1 3 3 2 1
\ / / / \ \
3 2 1 1 3 2
/ / \ \
2 1 2 3
——题目难度:中等
前言
二叉查找树(Binary Search Tree),(又:二叉搜索树,二叉排序树)它或者是一棵空树,或者是具有下列性质的二叉树: 若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值; 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值; 它的左、右子树也分别为二叉搜索树。二叉搜索树作为一种经典的数据结构,它既有链表的快速插入与删除操作的特点,又有数组快速查找的优势;所以应用十分广泛,例如在文件系统和数据库系统一般会采用这种数据结构进行高效率的排序与检索操作。
分析
看到题目要求要统计数目 应该能想到使用动态规划解题。
可以用dp[i]来保存记录:当有 i 个节点时所能组成的二叉搜索树的个数,令 f(j) 以 j 为根的二叉搜索树的个数
那么 dp(i) = f(1) + f(2) + f(3) + ... + f(i)
又当 j 为根节点时,其左子树节点个数为 j - 1 个,右子树节点数为 i - j 个,
则 f(j) = dp(j - 1) * dp(i - j)
所以 有如下公式:
dp(i) = dp(0) * dp(i - 1) + dp(1) * dp(i - 2) + ... + dp(i - 1) * dp(0)
最后返回 dp(n) 即可。
-下面代码
class Solution {
public:
int numTrees(int n) {
/*dp[i]表示长度为 i 的序列能构成的不同二叉搜索树的个数*/
vector<int> dp(n+1);
dp[1] = dp[0] = 1;
for(int i = 2; i <= n; i++) { //这层循环用来求长度为 i 的序列的不同二叉搜索树的个数
for(int j = 1; j <= i; j++) { //这层循环用来累加 以 j 为根、序列长度为 i 的不同二叉搜索树的个数
dp[i] += dp[j-1] * dp[i-j];
}
}
return dp[n];
}
};
结果