感觉更像是一个公式推导题,具体就直接看题解吧,推导挺简单的:不同的二叉搜索树 - 不同的二叉搜索树 - 力扣(LeetCode)
动态规划
一个重点:注意到 G(n) 代表长为i的序列能构成的二叉搜索树种数,它和序列的内容无关,只和序列的长度有关。
class Solution {
public:
int numTrees(int n) {
vector<int> G(n+1, 0);
G[0] = 1;
G[1] = 1; //G[i]代表长为i的序列能构成的二叉搜索树种数
for(int i = 2; i <= n; ++i){//i代表序列长度
for(int j = 1; j <= i; ++j){//j代表以j为根节点
G[i] += G[j-1]*G[i-j];
}
}
return G[n];
}
};
如果没想到上述的推导的话,可以直接用递归,理解起来也容易,不同的二叉搜索树 - 力扣(LeetCode)
f ( n ) = f ( 0 ) ∗ f ( n − 1 ) + f ( 1 ) ∗ f ( n − 2 ) + . . . + f ( n − 1 ) ∗ f ( 0 ) f(n)=f(0)*f(n-1)+f(1)*f(n-2)+...+f(n-1)*f(0) f(n)=f(0)∗f(n−1)+f(1)∗f(n−2)+...+f(n−1)∗f(0)
递归
class Solution {
public:
int numTrees(int n) {
if(n == 0) return 1;
if(n <3) return n;
int sum = 0;
for(int i = 0; i < n; ++i)
sum += numTrees(i) * numTrees(n-i-1);
return sum;
}
};
上述递归的效率并不高,因为存在着大量的重复运算,所以明显就是要用记忆化递归了:
class Solution {
public:
vector<int> vec;
int numTrees(int n) {
if(vec.size() == 0){
vec.resize(30);
}
if(n == 0) return 1;
if(n < 3) return n;
int sum = 0;
for(int i = 0; i < n; ++i){
if(vec[i] == 0)
vec[i] = numTrees(i);
if(vec[n-i-1] == 0)
vec[n-i-1] = numTrees(n-i-1);
sum += vec[i]*vec[n-i-1];
}
return sum;
}
};
或者这么写:
class Solution {
public:
vector<int> vec;
int numTrees(int n) {
if(vec.size() == 0){
vec.resize(n+1, 0);
vec[0] = 1;
}
if(vec[n] != 0) return vec[n];
int sum = 0;
for(int i = 0; i < n; ++i)
sum += numTrees(i) * numTrees(n-i-1);
vec[n] = sum;
return sum;
}
};