问题描述:
给定一个整数 n,求以 1 … n 为节点组成的二叉搜索树有多少种?
二叉查找树(Binary Search Tree),(又:二叉搜索树,二叉排序树)它或者是一棵空树,或者是具有下列性质的二叉树: 若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值; 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值; 它的左、右子树也分别为二叉排序树。
思路分析:
假设以 1 … n 为节点组成的二叉搜索树的个数为 dp[n]
根据二叉搜索树的性质可知,假设某棵由[1,2…n]构成的二叉搜索树的根节点为数字i,则其左子树一定是由[1,2…i-1]构成的二叉搜索树(dp[i-1]),其右子树一定是由[i+1,…,n]构成的二叉搜索树。
假设[1,2…i-1]构成的二叉搜索树有m种,由[i+1,…,n]构成的二叉搜索树有n种,则以i为根节点由[1,2…n]构成的二叉搜索树共有m * n种。
又因为 由[i+1,…,n]构成的二叉搜索树 与 由[1,…,n-i]构成的二叉搜索树(dp[n-i])是一样的(它们数的大小关系是一样的,都是连续的n-i个数)。
所以,以i为根节点由[1,2…n]构成的二叉搜索树的个数,可以分解为两个子问题(dp[i-1], dp[n-i])的乘积。
而以 1 … n 为节点组成的二叉搜索树可以按照根节点的数字,划分为n中不同的情况(根节点为1,2,3…)。所以dp[n] 为n中情况的个数之和。
代码:
class Solution {
public:
int numTrees(int n) {
int dp[n+1];
memset(dp, 0, sizeof(dp));
dp[0] = 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];
}
};