![](https://img-blog.csdnimg.cn/img_convert/4c90fed943f76d0518c2da19ebc65660.png)
完全二叉树是一种每个节点都有2个分支或0个分支,但从来没有1个分支的树。
编写一个函数,返回恰好有n个叶结点的唯一完整二叉树结构的数目。
对于那些对组合学感兴趣的人来说,这个问题确实有一个封闭形式的解决方案):
原文题目提示里面给了个维基百科的关于卡特兰数的链接,那么说明这题肯定是要用卡塔兰数来求解的,这里附上理解,不过好像要科学上网才可以进去。卡塔兰数 - 维基百科,自由的百科全书 (wikipedia.org)
说实话他的推导还有性质啥的我一点也看不懂,但是在应用那里可以看到它能用来求解n个节点组成不不同构二叉树的方案数。
![](https://img-blog.csdnimg.cn/img_convert/d3fec3d3066d59e03e47d4d3e8bbd713.png)
那我们就知道卡特兰数C_n表示在有n个节点的二叉树中,可以构建的不同形态的二叉树的数目。
这个递归关系式的意思是,有n+1个节点的二叉树,可以通过以下方式构建:
将1个节点放在左子树中,剩下n-1个节点放在右子树中,这样可以构建C_{n}个不同形态的二叉树;
将2个节点放在左子树中,剩下n-2个节点放在右子树中,这样可以构建C_{n-1}个不同形态的二叉树;
...
将n-1个节点放在左子树中,剩下1个节点放在右子树中,这样可以构建C_0个不同形态的二叉树。
将这些情况的结果累加起来,就可以得到有n+1个节点的二叉树的不同形态的数目,即C_{n+1}。
(注意,根节点占一个节点的数量,并且根节点要么没有子节点要么就只能有两个子节点)
有了这个思路就很好解决问题了,我们直接来看代码。
def num_trees(n):
if n == 1 or n == 2:
return 1
result=0
for i in range(1, n):
result+=num_trees(i) * num_trees(n - i)
return result
这个函数使用递归的方式计算结果。如果n∈(1,2),说明只有一种结构,直接返回1。否则,遍历从1到n-1的所有可能左子树和右子树的组合,将它们的结构数相乘,再将所有可能的组合的结果相加,就是恰好有n个叶结点的唯一完整二叉树结构的数目。