【JZOJ5068】【GDSOI2017第二轮模拟】树

Description

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

Data Constraint

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

Solution

这道题是一道带标号的无根树计数，我们考虑prufer序列。
首先先讲一下prufer序列怎么求：对于一棵树，我们每次删去最大的叶子结点，并输出它的父亲，知道树中只剩下两个点，因为这时我们已经知道整棵树的连边情况，所以就不用删了，整个序列长度为n-2。求出一个prufer序后，我们就可以知道整棵树的连边情况，也就说树唯一确定。
我们设出f[i][j][k]表示前i个点选出j个点，构成的prufer序的长度为k（解释一下为什么会有k这一维：因为一开始有可能是森林，后面才逐渐合并成树）那么我们发现一个度数为a[i]的点最多在prufer中出现a[i]-1次，所以我们转移为f[i][j+1][k+l]+=f[i][j][k](0 <=l< a[i])，f[i][j][k]+=f[i-1][j][k]。复杂度O（n^4）.

Code

#include<iostream>#include<cmath>#include<cstring>#include<cstdio>#include<algorithm>#define ll long longusing namespace std;const ll maxn=1e2+5,mo=1004535809;ll f[maxn][maxn][maxn],a[maxn],c[maxn][maxn];ll n,i,t,j,k,l,x,y,z;int main(){    freopen("tree.in","r",stdin);freopen("tree.out","w",stdout);    scanf("%lld",&n);    for (i=1;i<=n;i++)        scanf("%lld",&a[i]);    c[0][0]=1;    for (i=1;i<=n;i++){        c[i][0]=1;        for (j=1;j<=n;j++)            c[i][j]=(c[i-1][j-1]+c[i-1][j])%mo;    }    f[0][0][0]=1;    for (i=1;i<=n;i++)        for (j=0;j<=i;j++)            for (k=0;k<n-1;k++){                f[i][j][k]=f[i-1][j][k];                if (!j) continue;                for (l=0;l<=min(k,a[i]-1);l++)                    f[i][j][k]=(f[i][j][k]+f[i-1][j-1][k-l]*c[k][l])%mo;            }    printf("%lld ",n);    for (i=2;i<=n;i++)        printf("%lld ",f[n][i][i-2]);}