Leetcode 96. 不同的二叉搜索树
题目
给定一个整数 n,求以 1 … n 为节点组成的二叉搜索树有多少种?
测试样例
输入: 3
输出: 5
解释:
给定 n = 3, 一共有 5 种不同结构的二叉搜索树:
1 3 3 2 1
\ / / / \ \
3 2 1 1 3 2
/ / \ \
2 1 2 3
题解
动态规划
令dp[i]表示i个节点组成的二叉搜索树的数目
根据二叉搜索树的特点,左子树的节点都小于根节点,右子树的节点都大于根节点。因此构造的过程中,我们遍历每个点充当根节点,假设我们现在有i个节点,我们让j+1作为根节点,则小于j+1的j个节点构成的二叉搜索树便成为了左子树;j+2~i的i-j-1个节点构成的二叉搜索树便成为了右子树。二者组合便成了所有的二叉搜索树,显然,我们的动态转移方程为
- dp[i] += dp[j]*dp[i-j-1],其中j>=0 && j<i
初始条件就是1、2个节点时,都只有一种二叉搜索树,即dp[0]=dp[1]=1
代码
int numTrees(int n) {
if(n == 0) return 0;
vector<int> dp(n+1,0);
dp[0]=1;
dp[1]=1;
for(int i=2; i<=n; i++){
for(int j=0; j<i; j++){
dp[i] += dp[j]*dp[i-j-1];
}
}
return dp[n];
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/unique-binary-search-trees
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。