（Prufer序列）洛谷P2624 [HNOI2008]明明的烦恼

洛谷P2624 [HNOI2008]明明的烦恼

思路：

设要求度数的点有$k$个，$sum=\sum _{i=1}^k(d_i-1)$。
根据Prufer序列。
如果$d_i=0$或$sum>n-2$都是不合法的。
对于已知的$sum$个点，方案数就是求全排列，即：
$$\frac{sum!}{\prod _{i=1}^k(d_i-1)!}$$
又因为这$sum$个点是从$n-2$个点中选的，所以要乘上组合数，即：
$$\left(\genfrac{}{}{0ex}{}{n-2}{sum}\right)\frac{sum!}{\prod _{i=1}^k(d_i-1)!}$$
剩下的$n-2-sum$可以从其余的$n-k$个点中任选，即：
$$\left(\genfrac{}{}{0ex}{}{n-2}{sum}\right)\frac{sum!}{\prod _{i=1}^k(d_i-1)!}(n-k{)}^{n-2-sum}$$
化简得：
$$\frac{(n-2)!}{(n-2-sum)!\prod _{i=1}^k(d_i-1)!}(n-k{)}^{n-2-sum}$$
看数据范围，直接用python就行。

代码：

n=(int)(input())
k=0;sum=0
a=[0]*1100
f=[0]*1100
f[0]=1;
for i in range(1,1001):
    f[i]=f[i-1]*i;
for i in range(1,n+1):
    a[i]=(int)(input())
    if(a[i]==0):
        print(0)
        exit()
    if(a[i]!=-1):
        sum+=a[i]-1
        k+=1
if(sum>n-2):
    print(0)
    exit()
ans=f[n-2]//f[n-2-sum]
for i in range (1,n+1):
    if(a[i]!=-1):
        ans//=f[a[i]-1]
ans*=pow(n-k,n-2-sum)
print(ans)