1-1 T1 tree(dp+组合数学)

题目描述

有 n 个点,第 i 个点的限制为度数不能超过 ai。
现在对于每一个 s(1<=s<=n)
问从这 n 个点中选出 s 个点组成有标号无根树的方案数。
答案对1004535809取模。

输入输出格式

输入格式:

第一行一个整数表示n。
第二行n个整数a1~an。

输出格式:

一个n个整数,第i个整数表示s=i时的答案。

输入输出样例

输入样例#1: 复制
3
2 2 1
输出样例#1: 复制
3 3 2

说明

对于20%的数据,n≤6。
对于60%的数据,n≤50。
对于100%的数据,n≤100。


首先树的计数问题,肯定是prufer序列辣
然后发现好像是个dp?
然后写了个状态:

dp[i][j][k]ijpruferk
然后回想一下原来有个题
大概就是说知道每个点的度数,然后问无根树的数量
可以这样计算:
ans=(n2)!ni=1(di1)!
然后就是把这个东西转换成dp就好了
我们发现下面那一堆都好处理,就是上面那个东西不好处理
我们可以先把那个东西拿出来,最后再乘回去就好了
根据原来那个dp状态,就可以写转移方程了:
dp[i+1][j][k]+=dp[i][j][k]
dp[i+1][j+1][k+l]+=dp[i][j][k]1l!(l[0,Ai1])
然后直接dp就好了
我们可以枚举每个点在prufer序列中的出现次数
然后虽然理论上界是 O(n4) 但是加几个小优化,就远远跑不到,单点最多132ms
那些什么阶乘和逆元都可以预处理。。
代码:

// luogu-judger-enable-o2
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<iostream>
#include<algorithm>
#define ll long long
#define max(a,b) a>b?a:b
#define min(a,b) a<b?a:b
using namespace std;
inline int read(){
    int x=0,f=1;char ch=' ';
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}
const int N=105,p=1004535809;
int n;
int d[N],dp[N][N][N];
ll fac[N],ifac[N];
inline ll ksm(ll a,ll b){
    ll ans=1;
    while(b){
        if(b&1)ans=(ans*a)%p;
        a=(a*a)%p;
        b>>=1;
    }
    return ans;
}
inline void init(){
    fac[0]=1;
    for(int i=1;i<=n;++i)fac[i]=(fac[i-1]*i)%p;
    ifac[n]=ksm(fac[n],p-2);ifac[0]=1;
    for(int i=n-1;i>=1;--i)ifac[i]=(ifac[i+1]*(i+1))%p;
}
int main(){
    n=read();init();
    for(int i=0;i<n;i++)d[i]=read();
    dp[0][0][0]=1;
    for(int i=0;i<n;++i){
        for(int j=0;j<n;++j){
            for(int k=0;k<n;++k){
                if(!dp[i][j][k])continue;
                dp[i+1][j][k]+=dp[i][j][k];
                if(dp[i+1][j][k]>=p)dp[i+1][j][k]-=p;
                for(int l=0;l<d[i];++l){
                    if(k+l>n-2)break;
                    dp[i+1][j+1][k+l]+=(dp[i][j][k]*ifac[l])%p;
                    if(dp[i+1][j+1][k+l]>=p)dp[i+1][j+1][k+l]-=p;
                }
            }
        }
    }
    printf("%d ",n);
    for(int s=2;s<=n;++s){
        int ans=(dp[n][s][s-2]*fac[s-2])%p;
        printf("%d ",ans);
    }
    return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值