BZOJ1211 - [HNOI2004]树的计数

Portal

Description

一个有\(n(n\leq150)\)个结点的树，给出每个节点的度数\(d_i\)，求满足度数要求的无根树的个数。

Solution

prufer数列_百度百科
简单来说，一棵\(n\)个节点的无根树与一个数值在\([1,n]\)的长度为\(n-2\)的序列一一对应，这个序列便称作prufer数列。其中若点\(i\)的度数为\(d_i\)，那么\(i\)在prufer数列中就出现\(d_i-1\)次。
那么这道题就很简单了：向一个长度为\(n-2\)的序列中填入\(d_1-1\)个\(1\)，\(d_2-1\)个\(2\)，...，\(d_n-1\)个\(n\)。
\[\begin{align*} ans &= \binom{n-2}{d_1-1} \binom{n-2-(d_1-1)}{d_2-1} ...\binom{n-2-\sum_{i=1}^{n-1}(d_i-1)}{d_n-1} \\ &= \frac{(n-2)!}{(d_1-1)!(n-2-(d_1-1))!} \cdot \frac{(n-2-(d_1-1))!}{(d_2-1)!(n-2-\sum_{i=1}^2(d_i-1))} ... \frac{(n-2-\sum_{i=1}^{n-1}(d_i-1))!}{(d_n-1)!0!} \\ &= \frac{(n-2)!}{\prod_{i=1}^n (d_i-1)!} \end{align*}\]由于阶乘比较大而且坑爹的不取模，所以通过分解质因数的方法来计算。

时间复杂度\(O(n^2)\)。

Code

//[HNOI2004]树的计数
#include <cstdio>
int const N=200;
int n,d[N];
int fac[N][N];
void init()
{
    for(int i=1;i<=n;i++)
    {
        int x=i;
        for(int j=2;j<=n;j++)
            while(x%j==0) fac[i][j]++,x/=j;
    }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++) fac[i][j]+=fac[i-1][j];
}
int ansP[N];
int main()
{
    scanf("%d",&n); init();
    int sum=0;
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&d[i]); sum+=d[i];
        if(d[i]==0&&n!=1) {puts("0"); return 0;}
    }
    if(sum!=n*2-2) {puts("0"); return 0;}
    if(n==1) {puts("1"); return 0;}
    for(int i=1;i<=n;i++) ansP[i]=fac[n-2][i];
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++) ansP[j]-=fac[d[i]-1][j];
    long long ans=1;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=ansP[i];j++) ans*=i;
    printf("%lld\n",ans);
    return 0;
}

P.S.

注意要特判掉无解和\(n=1\)的情况。

转载于:https://www.cnblogs.com/VisJiao/p/BZOJ1211.html