给出二叉树每个节点的深度,请你计算满足条件的二叉树有多少种
我没有写输入输出语句,可自行补上。
主要思路:除根节点外,其他每个节点所在的层可能出现的情况都由上一层节点个数决定。将每一层的情况相乘得到总数。
// 代码如下
# 导入scipy的排列组合函数
from scipy.special import comb
a = 4
b = [1, 0, 2, 2]
val_count = {}
b.sort()
for val in b:
val_count[val] = val_count.get(val, 0) + 1
typ = 1
for key in list(val_count.keys())[1:]:
typ = typ * int(comb(2*val_count.get(key-1), val_count.get(key)))
print(typ % (10**9 + 7))