[AGC005F] Many Easy Problem

XXXII.[AGC005F] Many Easy Problems

直接计算恐怕不太容易,正难则反,考虑一个节点何时不被包含在一个连通块内。

显然,假如我们以当前节点为根,则当且仅当集合中所有节点同处在其某一个儿子的子树内,当前节点不在连通块内。

我们设 \(f(i,j)\) 表示节点 \(i\) 对大小为 \(j\) 的集合不被包含的数量。则有

\[f(i,j)=\sum\limits_{\text{边}(i,i')}\dbinom{i'\text{的子树大小}}{j} \]

\[ans_j=n\dbinom{n}{j}-\sum\limits_{i=1}^nf(i,j) \]

考虑令 \(cnt_i\) 表示大小为 \(i\) 的子树数量(当然,是无根树的子树,即树上一条边所连接着的两坨节点的数量)。则有

\[ans_j=n\dbinom{n}{j}-\sum\limits_{i}\dbinom{i}{j}cnt_i \]

后面的东西随随便便拆个组合数然后瞎卷一下即可。

时间复杂度 \(O(n\log n)\)

代码:

#include<bits/stdc++.h>
using namespace std;
const int N=1<<20;
const int mod=924844033;
const int G=5;
int rev[N];
int ksm(int x,int y=mod-2){
	int z=1;
	for(;y;y>>=1,x=1ll*x*x%mod)if(y&1)z=1ll*z*x%mod;
	return z;
}
void NTT(int *a,int tp,int LG){
    int lim=(1<<LG),invlim=ksm(lim,mod-2);
    for(int i=0;i<lim;i++)rev[i]=(rev[i>>1]>>1)|((i&1)<<(LG-1));
    for(int i=0;i<lim;i++)if(i<rev[i])swap(a[i],a[rev[i]]);
    for(int md=1;md<lim;md<<=1){
        int rt=ksm(G,(mod-1)/(md<<1));
        if(tp==-1)rt=ksm(rt,mod-2);
        for(int stp=md<<1,pos=0;pos<lim;pos+=stp){
            int w=1;
            for(int i=0;i<md;i++,w=(1ll*w*rt)%mod){
                int x=a[pos+i],y=(1ll*w*a[pos+md+i])%mod;
                a[pos+i]=(x+y)%mod;
                a[pos+md+i]=(x-y+mod)%mod;
            }
        }
    }
    if(tp==-1)for(int i=0;i<lim;i++)a[i]=(1ll*a[i]*invlim)%mod;
}
int A[N],B[N];
void mul(int *a,int *b,int *c,int LG){//using: Array A and B
    int lim=(1<<LG);
    for(int i=0;i<lim;i++)A[i]=B[i]=0;
    for(int i=0;i<(lim>>1);i++)A[i]=a[i],B[i]=b[i];
    NTT(A,1,LG),NTT(B,1,LG);
    for(int i=0;i<lim;i++)A[i]=1ll*A[i]*B[i]%mod;
    NTT(A,-1,LG);
    for(int i=0;i<lim;i++)c[i]=A[i];
}
int n,cnt[N],fac[N],inv[N],f[N],g[N];
vector<int>v[N];
int dfs(int x,int fa){
	int sz=1;
	for(auto y:v[x]){
		if(y==fa)continue;
		int zs=dfs(y,x);
		cnt[zs]++,cnt[n-zs]++;
		sz+=zs;
	}
	return sz;
}
int C(int x,int y){return 1ll*fac[x]*inv[y]%mod*inv[x-y]%mod;}
int main(){
	scanf("%d",&n);
	for(int i=1,x,y;i<n;i++)scanf("%d%d",&x,&y),v[x].push_back(y),v[y].push_back(x);
	dfs(1,0);
	fac[0]=1;for(int i=1;i<=n;i++)fac[i]=1ll*fac[i-1]*i%mod;
	inv[n]=ksm(fac[n]);for(int i=n-1;i>=0;i--)inv[i]=1ll*inv[i+1]*(i+1)%mod;
//	for(int i=0;i<=n;i++)printf("%d ",fac[i]);puts("");
//	for(int i=0;i<=n;i++)printf("%d ",inv[i]);puts("");
	for(int i=0;i<=n;i++)f[n-i]=1ll*fac[i]*cnt[i]%mod,g[i]=inv[i];
	int all=0;
	while((1<<all)<=n)all++;
	mul(f,g,f,all+1);
	for(int i=1;i<=n;i++)printf("%d\n",(1ll*n*C(n,i)%mod+mod-1ll*f[n-i]*inv[i]%mod)%mod);
	return 0;
} 

原文地址:https://www.cnblogs.com/Troverld/p/14608111.html