题目描述:
给定一个整数 n,求以 1 … n 为节点组成的二叉搜索树有多少种?
示例:
输入: 3
输出: 5
解释:
给定 n = 3, 一共有 5 种不同结构的二叉搜索树:
方法1 动态规划
主要思路:
(1)对于给定的序列1,2,……,n,求其组成的搜索树的数量,遍历每个每一个序列号 i,则 i 左边的序列1,2,……i-1和右边的 i+1,i+2,……,n的两组子序列,可以分别求取其可能组成的搜索树数量,并将左右数量相乘,既为该序号 i 下,可能组成的搜索树的数量,将所有的 i 遍历,相加就为n个元素下,可以组成的总的搜索树的数量,这就相当于把n分解成了子任务,进行求取;
(2)对于n=0或n=1,表示可以组成的搜索树的数量为1;
class Solution {
public:
int numTrees(int n) {
//初始化,处理特殊的情形
if(n<2)
return 1;
if(n==2)
return 2;
//声明动态规划的存储数组
vector<int> dp(n+1,0);
//对特殊的情形进行初始化
dp[0]=1;
dp[1]=1;
dp[2]=2;
for(int i=3;i<=n;++i){
//从 i 的左右两侧开始分组
//表示 i 的左边有left个元素的情形
int left=0;
//表示 i 的右边有right个元素的情形
int right=i-1;
//遍历满足要求的情形
while(right>left){
dp[i]+=dp[left++]*dp[right--];
}
//重复的直接乘以2
dp[i]*=2;
//根据奇偶性,左右两边的分组情形可能存在数量相等的情形,单独处理
if(right==left)
dp[i]+=dp[left]*dp[right];
}
//返回结果
return dp[n];
}
};