题目
原题链接
给定一个整数 n,求以 1 … n 为节点组成的二叉搜索树有多少种?
示例:
输入: 3
输出: 5
解释:
给定 n = 3, 一共有 5 种不同结构的二叉搜索树:
1 3 3 2 1
\ / / / \ \
3 2 1 1 3 2
/ / \ \
2 1 2 3
思路
这道题我们用DP解
设 dp[i]
表示用 i
个节点可以建出的树的个数,那么初始值就是:
dp[0] = 1
——空树只有一种情况dp[1] = 1
——只有一个根节点dp[2] = 2
——两个节点,两种情况
接下来,递推式呢?
N
个节点,我们分 N
种情况讨论,分别是以第 1…N 个节点为根的情况,这时候 dp[N]
的值就是 dp[左子树的值] * dp[右子树的值]
。也就是
dp[i] = dp[j] * dp[i - j - 1] {j = 1...N}
递推式就很愉快地被推出来了。
以下是代码:
class Solution {
public:
int numTrees(int n) {
vector<int> dp;
dp.push_back(1); dp.push_back(1); dp.push_back(2);
for(int i = 3; i <= n; ++ i) {
dp.push_back(0);
for(int j = 0; j < i; ++ j) {
dp[i] += dp[j] * dp[i - j - 1];
}
}
return dp[n];
}
};