记录一道关于树的算法题-->leetcode.96
题目:
给你一个整数 n
,求恰由 n
个节点组成且节点值从 1
到 n
互不相同的 二叉搜索树 有多少种?
返回满足题意的二叉搜索树的种数。
分析:
二叉搜索树:要满足搜索二叉树的性质,其根节点root的值要大于左节点的值,且要小于右节点的值。root.left.value<root.value<root.right.value
一棵树的排列方式有多少种:跟当前根节点元素,其左子树的排列种树,右子树的排列种树有关;
例如:有[1,2,3,4,5],5个节点,其值分别为1,2,3,4,5
因为不同的元素当根节点,是不同的排列组合
所以推导一下:
如果让“1”当根节点,按照二叉搜索树的性质,其左子树有0个节点,其右子树有[2,3,4,5]4个节点;
如果让“2”当根节点,按照二叉搜索树的性质,其左子树有[1]个节点,其右子树有[3,4,5]3个节点;
如果让“3”当根节点,按照二叉搜索树的性质,其左子树有[1,2]个节点,其右子树有[4,5]4个节点;
如果让“4”当根节点,按照二叉搜索树的性质,其左子树有[1,2,3]个节点,其右子树有[5]4个节点;
如果让“5”当根节点,按照二叉搜索树的性质,其左子树有[1,2,3,4]个节点,其右子树有0个节点;
如果记:f(n)表示,以当前n号元素为根节点的树,有多少种排列方式;
那么f(n)=?
以3为根节点为例:
f(3)=左节点个数的排列组合X右节点的排列组合=F(2)*F(2)=4;
其左右子树都可以看成是,只有2个节点的二叉搜索树的排列问题;
很明显这是条件一样的个子问题,上动态规划
当节点个数为n时,用F(n)表示所有求的种数
无可置疑:就有 F(0)=0;F(1)=1;F(2)=2;
则:F(n)=f(1)+f(2)+f(3)+......f(n); 每个节点轮流当根节点的总和
F(5):节点个数为5的树,总共有多少种排列方式:
F(5)=f(1)+f(2)+f(3)+f(4)+f(5)
f(1)=F(0)*F(4) :
如果让“1”当根节点,按照二叉搜索树的性质,其左子树有0个节点,其右子树有[2,3,4,5]4个节点;
f(2)=F(1)*F(3);
f(3)=F(2)*F(2);
f(4)=F(3)*F(1);
f(5)=F(4)*F(0);
F(5)=F(0)*F(4)+F(1)*F(3)+F(2)*F(2)+F(3)*F(1)+F(4)*F(0)
用代码表示:
class Solution {
public int numTrees(int n) {
int[] dp=new int[n+1];
//设置边界,当节点数为,0/1时,只有一种排序方式的树
dp[0]=1;
dp[1]=1;
for(int i=2;i<=n;i++)//i表示当前选用第几个元素,当根节点
{
for(int j=1;j<=i;j++)//j表示从f(1)+...f(n),这里的n就是i
{
dp[i]+=dp[j-1]*dp[i-j];
}
}
return dp[n];
}
}
扩展:求n个节点的二叉树排列
没有了二叉搜索树的限制,那么轮流挑选元素当根节点,就不需要了。
左右子树的个数,排列不同即可:
例如:F(5),左子树和右子树的个数,可以为(0,4)(1,3)(2,2)(3,1) (4,0)
F(n)=F(n-1)F(0) + F(n-2)F(1) + F(n-3)F(2) + ... + F(1)F(n-2) + F(n-1)F(0)