Unique Binary Search Trees(不同的二分查找树)
【难度:Medium】
Given n, how many structurally unique BST’s (binary search trees) that store values 1…n?
For example,
Given n = 3, there are a total of 5 unique BST’s.
解题思路
对于一个给定序列1……i……n(1<=i <=n),用该序列来构造BST,我们可以枚举i,用i来当做根节点root,那么自然地1……(i-1)就是root的左子树,(i+1)……n就是root的右子树。再对左右子树分别使用相同的方法求解,根据乘法原理,左子树的种类乘以右子树的种类就是以i为root节点所能构成的BST总个数,同时因为i唯一,所形成的BST也是不同的。
c++代码如下:
class Solution {
public:
int numTrees(int n) {
vector<int>v(n+1,0);
v[0] = 1;
//以i为root节点
for (int i = 1; i <= n; i++) {
//有i个数时,分别以j为子树root节点来构建BST的个数
for (int j = 1; j <= i; j++) {
v[i] += v[j-1]*v[i-j];
}
}
return v[n];
}
};