【jzoj5068】【GDSOI2017第二轮模拟】【树】【动态规划】

题目大意

有n个点,它们从1到n进行标号,第i个点的限制为度数不能超过A[i].

现在对于每个s (1 <= s <= n),问从这n个点中选出一些点组成大小为s的有标号无根树的方案数。

解题思路

首先我们要知道一棵带标号无根树的prufer序与树的形态一一对应。prufer序就是每次删掉度数最小的点并且把连出的边对应的点加入prufer序,剩下两个点时直接退出。还原时只需要找出最小没出现过的点,按顺序与prufer序上的点连边即可。

可以发现一个点在prufer序出现的次数加一就是它的度数,设长度为x的prufer序不同点出现的次数是b[i]那不同的序列有fact[x]/fact[b[i]],即每一个出现次数的阶乘都要除,设f[i][j][k]表示考虑到标号为i的点,选取了j个点,prufer序长度为k的方案数除以fact[x]的答案,f[i+1][j][k]+=f[i][j][k],f[i+1][j+1][k+l]+=f[i][j][k]/fact[l],其中0<=l<=a[i+1]-1。在最后输出答案的时候再乘上fact[x]即可。

code

#include
#include
#include
#include
#define LF double
#define LL long long
#define ULL unsigned long long
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
#define fr(i,j) for(int i=begin[j];i;i=next[i])
using namespace std;
int const mn=100+9,mo=1004535809;LL inf=1e9;
int n,a[mn];
LL fact[mn],ni[mn],f[mn][mn][mn];
int Pow(LL x,int y){
    LL z=1;
    while(y){
        if(y&1)z=z*x%mo;
        x=x*x%mo;
        y>>=1;
    }
    return z;
}
int main(){
    //freopen("tree.in","r",stdin);
    //freopen("tree.out","w",stdout);
    freopen("d.in","r",stdin);
    freopen("d.out","w",stdout);
    scanf("%d",&n);
    fo(i,1,n)scanf("%d",&a[i]);
    fact[0]=1;
    fo(i,1,n)fact[i]=fact[i-1]*i%mo;
    ni[n]=Pow(fact[n],mo-2);
    fd(i,n-1,0)ni[i]=ni[i+1]*(i+1)%mo;
    f[0][0][0]=1;
    fo(i,0,n-1)fo(j,0,n-1)fo(k,0,n)if(f[i][j][k]){
        f[i+1][j][k]=(f[i+1][j][k]+f[i][j][k])%mo;
        int tmp=min(n-k,a[i+1]-1);
        fo(l,0,tmp)
            f[i+1][j+1][k+l]=(f[i+1][j+1][k+l]+f[i][j][k]*ni[l])%mo;
    }
    printf("%d ",n);
    fo(i,2,n)printf("%lld ",f[n][i][i-2]*fact[i-2]%mo);
    return 0;
}

你可能感兴趣的:(动态规划,jzoj)