JZOJ 5068. 【GDSOI2017第二轮模拟】树

Description

有n个点，它们从1到n进行标号，第i个点的限制为度数不能超过A[i].
现在对于每个s (1 <= s <= n)，问从这n个点中选出一些点组成大小为s的有标号无根树的方案数。

Input

第一行一个整数n.
第二行n个整数表示A[i].

Output

输出一行n个整数，第i个整数表示s=i时的答案。答案模1004535809 = 479 * 2^{21} + 1。

Sample Input

3
2 2 1

Sample Output

3 3 2

Data Constraint

20%的数据：n <= 6
60%的数据：n <= 50
100%的数据：n <= 100

Solution

• 首先我们要知道一棵无根树的 prufer 数列与树的形态一一对应。

• 对于 prufer 数列不知道的可以查看 morejarphone 大佬关于 Prufer 数列的详细解释

• 对于大小为 s 的树，我们统计的是一个长度为 s−2 的序列。

• 而且可以发现一个点在 prufer 数列出现的次数加一就是它的度数。

• 所以度数不超过 Ai 即 i 在序列中的出现次数小于 Ai 。

• 若已知选出了 s 个点，且这些点的出现次数为 Ci ，则这些点组成一棵树的方案数为：

  s!C1!∗C2!∗…∗Cs!

Code

#include<cstdio>#include<cstring>#include<algorithm>#include<cctype>using namespace std;const int N=101,mo=1004535809;int a[N];long long f[N][N][N],g[N],h[N];inline int read(){    int X=0,w=0; char ch=0;    while(!isdigit(ch)) w|=ch=='-',ch=getchar();    while(isdigit(ch)) X=(X<<3)+(X<<1)+(ch^48),ch=getchar();    return w?-X:X;}inline int min(int x,int y){    return x<y?x:y;}inline long long ksm(long long x,int y){    long long s=1;    while(y)    {        if(y&1) s=s*x%mo;        x=x*x%mo;        y>>=1;    }    return s;}int main(){    int n=read();    g[0]=h[0]=f[0][0][0]=1;    for(int i=1;i<=n;i++) a[i]=read(),g[i]=g[i-1]*i%mo;    h[n]=ksm(g[n],mo-2);    for(int i=n-1;i;i--) h[i]=h[i+1]*(i+1)%mo;    for(int i=0;i<n;i++)        for(int j=0;j<=i;j++)            for(int k=0;k<=n;k++)                if(f[i][j][k])                {                    f[i+1][j][k]=(f[i+1][j][k]+f[i][j][k])%mo;//no                    for(int l=0,p=min(n-k-2,a[i+1]-1);l<=p;l++)                        f[i+1][j+1][k+l]=(f[i+1][j+1][k+l]+f[i][j][k]*h[l]%mo)%mo;//yes                }    printf("%d",n);    for(int i=2;i<=n;i++) printf(" %lld",f[n][i][i-2]*g[i-2]%mo);    return 0;}