zoukankan      html  css  js  c++  java
  • [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;
    } 
    

  • 相关阅读:
    es6扩展运算符深浅拷贝
    将工作慢慢移向python
    OpenGL
    翻译的一篇关于Normal Matrix (normal transformation)的文章
    顾城的诗
    复杂的C运行时库
    glut and glew
    setting boot up order in grub2
    Perlin Noise
    GL 与 CV 管线 (pipeline) 比较与相互转换
  • 原文地址:https://www.cnblogs.com/Troverld/p/14608111.html
Copyright © 2011-2022 走看看