nowcoder20037 [HNOI2004]树的计数

*ACoder*

于 2020-02-04 21:10:09 发布

阅读量152

点赞数

分类专栏： # 排列组合

本文链接：https://blog.csdn.net/FSAHFGSADHSAKNDAS/article/details/104175123

版权

排列组合专栏收录该内容

54 篇文章 0 订阅

订阅专栏

链接

点击跳转

题解

先判断是不是无解：无根树与 $p r u f e r$ 序列一一对应，所以合法的 $p r u f e r$ 序列存在的条件就是无根树存在的条件， $p r u f e r$ 序列存在的条件就是所有点的度数都大于 $0$ 而且都不大于 $n - 1$ ，而且所有点的度数之和要等于 $2 n - 2$

如果答案存在的话，那就用可重复排列组合公式：

$\frac{(n−2)!}{\prod^n_{i=1}(d_i−1)!}$

代码

import math
n = int(input())
lis = [int(x) for x in input().split()]
if n==1:
    if lis[0]==0:
        print(1)
    else:
        print(0)
    exit(0)
tot = sum(lis) - n
if tot != n-2:
    print(0)
    exit(0)
ans = math.factorial(n-2)
for d in lis:
    if d<=0 or d>n-1:
        print(0)
        exit(0)
    ans //= math.factorial(d-1)
print(ans)