思路:
DP
关键是要总结出二叉查找树个数的规律。
如果有0个元素,即空树,显然有1颗查找树;
f[0] = 1;
如果有1个元素,单节点树,显然也只有1颗查找树;
f[1] = 1;
如果有2个元素,那么有如下2种可能:
//左子树为0个元素的个数*右子树为1个元素的个数 + 左子树为1个元素的个数*右子树为0个元素的个数
f[2] = f[0]*f[1] +f[1]*f[0];
如果有3个元素,那么有如下5种可能:
//左子树为0个元素的个数*右子树为2个个元素的个数 + 左子树为1个元素的个数*右子树为1个元素的个数 + 左子树为2个元素的个数*右子树为0个元素的个数
f[3] = f[0]*f[2] + f[1]*f[1] + f[2]*f[0];
以此类推,可以得到递推公式:
AC代码如下:
class Solution {
public:
int numTrees(int n) {
vector<int> f(n+1,0);
if(n == 0) return 1;
f[0] = 1;
f[1] = 1;
for(int i = 2; i <= n; ++i) {
for(int k = 1; k <= i; ++k) {
f[i] += f[k - 1] * f[i - k];
}
}
return f[n];
}
};