题目
给定一个整数 n,求以 1 … n 为节点组成的二叉搜索树有多少种?
示例:
输入: 3
输出: 5
解释:
给定 n = 3, 一共有 5 种不同结构的二叉搜索树:
1 3 3 2 1
\ / / / \
3 2 1 1 3 2
/ / \
2 1 2 3
解答
首先是采用分治的思想:假设x个节点的不同搜索二叉树的种类为f(x)。那么n个节点中,根节点的选取有n种;与之相应的左右子树的节点数量对为:(0,n-1),(1,n-2),…,(n-1,0),那么f(n) = f(0)*f(n-1) + f(1)*f(n-2) + … + f(n-1)*f(0);
这是典型的卡特兰数问题,所以利用其中的公式:
K
(
n
)
=
C
2
n
n
−
C
2
n
n
−
1
K(n)=C^n_{2n}-C^{n-1}_{2n}
K(n)=C2nn−C2nn−1
然后求解组合数时,使用杨辉三角来进行求解。
class Solution {
public:
double C(int n, int k)
{
if (k == n || k == 0) return 1;
vector<double> f(k + 1, 1);
int cnt = 2;
while (cnt <= n)
{
int t = min(cnt-1, k);
for (int j = t; j >= 1; --j)
{
f[j] = f[j - 1] + f[j];
}
cnt++;
}
return f[k]-f[k-2];
}
int numTrees(int n)
{
if(n == 0) return 1;
if(n == 1) return 1;
return int(C(2*n-1,n));
}
};
作者:shuitang-2
链接:https://leetcode-cn.com/problems/unique-binary-search-trees/solution/qia-te-lan-shu-yang-hui-san-jiao-by-shuitang-2/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。