[AGC005F] Many Easy Problem
XXXII.[AGC005F] Many Easy Problems
直接计算恐怕不太容易,正难则反,考虑一个节点何时不被包含在一个连通块内。
显然,假如我们以当前节点为根,则当且仅当集合中所有节点同处在其某一个儿子的子树内,当前节点不在连通块内。
我们设 表示节点 对大小为 的集合不被包含的数量。则有
而
考虑令 表示大小为 的子树数量(当然,是无根树的子树,即树上一条边所连接着的两坨节点的数量)。则有
后面的东西随随便便拆个组合数然后瞎卷一下即可。
时间复杂度 。
代码:
#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;
}
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· go语言实现终端里的倒计时
· 如何编写易于单元测试的代码
· 10年+ .NET Coder 心语,封装的思维:从隐藏、稳定开始理解其本质意义
· .NET Core 中如何实现缓存的预热?
· 从 HTTP 原因短语缺失研究 HTTP/2 和 HTTP/3 的设计差异
· 分享一个免费、快速、无限量使用的满血 DeepSeek R1 模型,支持深度思考和联网搜索!
· 基于 Docker 搭建 FRP 内网穿透开源项目(很简单哒)
· ollama系列1:轻松3步本地部署deepseek,普通电脑可用
· 按钮权限的设计及实现
· 【杂谈】分布式事务——高大上的无用知识?