由于二叉搜索树的左右子树依然是二叉搜索树,所以考虑DP算法,即:
1. n=0 或 n=1时,个数为1;
2. n=2时,个数为2;
3. n=3时,确定根节点后分别计算左右子树的可能情况,然后相乘就是当前根节点下所有的变形种类,之后在求和即可。
算法如下:
class Solution {
public:
int numTrees(int n) {
if(n<=1) return 1;
vector<int> sum(n+1,0);
sum[0]=1;
sum[1]=1;
int i,j;
for(i=2;i<=n;i++)
{
for(j=0;j<i;j++)
{
sum[i]+=sum[j]*sum[i-1-j];
}
}
return sum[n];
}
};