96.不同的二叉搜索树
时间:2020年7月15日
知识点:卡特兰数
题目链接: https://leetcode-cn.com/problems/unique-binary-search-trees/
题目
给定一个整数 n,求以 1 … n 为节点组成的二叉搜索树有多少种?
示例1
输入:
3
输出:
5
说明:
给定 n = 3, 一共有 5 种不同结构的二叉搜索树:
1 3 3 2 1
\ / / / \ \
3 2 1 1 3 2
/ / \ \
2 1 2 3
解法
n = 1
1
n = 2
1 2
\ /
2 1
n = 3
1 1 2 3 3
\ \ / \ / /
2 3 1 3 2 1
\ / / \
3 2 1 2
左边节点0个右边节点2个 左边节点1个右边节1一个 左边节点2个右边节点0个
dp[]代表有几个节点的时候有几种方法。
- 左边节点或者右边节点个数为0的时候,dp[0]=1
- 左边节点或者右边节点个数为1的时候, dp[1]=1
- n=2时,有两种情况,
- 左边节点数为0,右边节点数=n-1-0(n-1,因为一个节点当根节点去了),总共dp[0]*dp[1]=1
- 左边节点数为1,右边节点数=n-1-1,总共dp[1]*dp[0]=1
两种情况相加dp[2]=2
- n=3时,有三种情况,
- 左边节点数为0,右边节点数=n-1-0,总共dp[0]*dp[2]=2
- 左边节点数为1,右边节点数=n-1-1,总共dp[1]*dp[1]=1
- 左边节点数为2,右边节点数=n-1-2,总共dp[2]*dp[0]=2
三种情况相加dp[3]=5
- n=4时,有四种情况,
- 左边节点数为0,右边节点数=n-1-0,总共dp[0]*dp[3]=5
- 左边节点数为1,右边节点数=n-1-1,总共dp[1]*dp[2]=2
- 左边节点数为2,右边节点数=n-1-2,总共dp[2]*dp[1]=2
- 左边节点数为3,右边节点数=n-1-3,总共dp[3]*dp[0]=5
四种情况相加dp[4]=15
- 由此可以得到 卡特兰数 公式
dp(n)=dp(0)∗dp(n−1)+dp(1)∗dp(n−2)+…+dp(n−1)∗dp(0) - 由此可以得出递推式
dp(0) = 1 dp(n+1)=2*(2n+1)/(n+2)*dp(n)
代码
#include <stdio.h>
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
int numTrees(int n) {
vector<int> dp(n+1,0);
dp[0] = 1;
dp[1] = 1;
for (int i=2;i<=n;i++){
for (int j=1;j<=i;j++){
dp[i]+=dp[j-1]*dp[i-j];
}
}
return dp[n];
}
};
int main()
{
//n=5 42 n=6 132
int n = 5;
Solution s ;
cout<<s.numTrees(n);
}
今天也是爱zz的一天哦!