[AT2064] [agc005_f] Many Easy Problems

题目链接

AtCoder:https://agc005.contest.atcoder.jp/tasks/agc005_f

洛谷:https://www.luogu.org/problemnew/show/AT2064

Solution

注意到模数为(441cdot 2^{21}+1),嘿嘿

首先要想到考虑贡献,然后这题就简单了。

设当前要算的为(f(i)),我们考虑第(x)个点的贡献,显然可以得到贡献为:

[inom{n}{i}-sum_{vin son_{x}}inom{sz_v}{i} ]

就是所有的方案减去这个点选不到的方案,其中(sz_v)表示以(x)为根(v)( m size)

然后累和:

[egin{align} f(i)&=sum_{x=1}^{n}left(inom{n}{i}-sum_{vin son_{x}}inom{sz_v}{i} ight)\ &=ninom{n}{i}-sum_{x=1}^{n}sum_{vin son_x}inom{sz_v}{i} end{align} ]

后面那块不好处理,我们可以记个桶(cnt(s))表示大小为(s)的子树出现了多少次,这个一遍( m dfs)就可以算出来。

那么把式子化一下:

[egin{align} f(i)&=ninom{n}{i}-sum_{j=1}^{n}cnt(j)inom{j}{i}\ &=ninom{n}{i}-frac{1}{i!}sum_{j=i}^{n}frac{cnt(j)j!}{(j-i)!} end{align} ]

注意到后面可以转化为卷积的形式,设:

[g(i)=cnt(i)i!,h(i)=frac{1}{i!} ]

后面的(sum)就是:

[sum_{j=i}^{n}g(j)h(j-i)=sum_{j=0}^{n-i}g(j-i)h(j) ]

我们( m reverse)一下(g)

[sum_{j=0}^{n-i}g_R((n-i)-j)h(j) ]

(NTT)优化就好了,复杂度(O(nlog n))

注意这个神奇的模数原根是(5),我一开始写成(3)还以为是(NTT)挂了...

#include<bits/stdc++.h>
using namespace std;

void read(int &x) {
    x=0;int f=1;char ch=getchar();
    for(;!isdigit(ch);ch=getchar()) if(ch=='-') f=-f;
    for(;isdigit(ch);ch=getchar()) x=x*10+ch-'0';x*=f;
}

void print(int x) {
    if(x<0) putchar('-'),x=-x;
    if(!x) return ;print(x/10),putchar(x%10+48);
}
void write(int x) {if(!x) putchar('0');else print(x);putchar('
');}

#define lf double
#define ll long long
#define end puts("NO"),exit(0)

const int maxn = 1e6+10;
const int inf = 1e9;
const lf eps = 1e-8;
const int mod = 924844033;

int add(int x,int y) {return x+y>mod?x+y-mod:x+y;}
int del(int x,int y) {return x-y<0?x-y+mod:x-y;}
int mul(int x,int y) {return 1ll*x*y-1ll*x*y/mod*mod;}

int qpow(int a,int x) {
    int res=1;
    for(;x;x>>=1,a=mul(a,a)) if(x&1) res=mul(a,res);
    return res;
}

int fac[maxn],ifac[maxn],inv[maxn],pos[maxn],N,mxn,bit,n;
int w[maxn],rw[maxn],f[maxn],g[maxn],h[maxn],head[maxn],tot,sz[maxn];
struct edge{int to,nxt;}e[maxn<<1];

void ins(int u,int v) {e[++tot]=(edge){v,head[u]},head[u]=tot;}

void dfs(int x,int fa) {
    sz[x]=1;
    for(int i=head[x];i;i=e[i].nxt) 
        if(e[i].to!=fa) dfs(e[i].to,x),sz[x]+=sz[e[i].to],g[sz[e[i].to]]++;
    g[n-sz[x]]++;
}

void ntt_init() {
    w[0]=rw[0]=1,w[1]=qpow(5,(mod-1)/mxn);
    for(int i=2;i<=mxn;i++) w[i]=mul(w[i-1],w[1]);
    rw[1]=qpow(w[1],mod-2);
    for(int i=2;i<=mxn;i++) rw[i]=mul(rw[i-1],rw[1]);
}

void ntt_get(int len) {
    for(bit=0,N=1;N<=len;N<<=1,bit++);
    for(int i=0;i<N;i++) pos[i]=pos[i>>1]>>1|((i&1)<<(bit-1));
}

void ntt(int *r,int op) {
    for(int i=1;i<N;i++) if(pos[i]>i) swap(r[i],r[pos[i]]);
    for(int i=1,d=mxn>>1;i<N;i<<=1,d>>=1) 
        for(int j=0;j<N;j+=i<<1) 
            for(int k=0;k<i;k++) {
                int x=r[j+k],y=mul((op==-1?rw:w)[k*d],r[i+j+k]);
                r[j+k]=add(x,y),r[i+j+k]=del(x,y);
            }
    if(op==-1) {int d=qpow(N,mod-2);for(int i=0;i<N;i++) r[i]=mul(r[i],d);}
}

int main() {
    read(n);for(int x,y,i=1;i<n;i++) read(x),read(y),ins(x,y),ins(y,x);
    dfs(1,0);fac[0]=ifac[0]=inv[0]=inv[1]=1;
    for(int i=1;i<=n;i++) fac[i]=mul(fac[i-1],i);
    for(int i=2;i<=n;i++) inv[i]=mul(mod-mod/i,inv[mod%i]);
    for(int i=1;i<=n;i++) ifac[i]=mul(ifac[i-1],inv[i]);
    for(int i=0;i<=n;i++) h[i]=ifac[i],g[i]=mul(g[i],fac[i]);
    g[0]=0;
    
    reverse(g,g+n+1);for(mxn=1;mxn<=n<<1;mxn<<=1);
    ntt_init(),ntt_get(n<<1),ntt(g,1),ntt(h,1);
    for(int i=0;i<N;i++) f[i]=mul(g[i],h[i]);
    ntt(f,-1);
    for(int i=1;i<=n;i++) write(del(mul(n,mul(fac[n],mul(ifac[i],ifac[n-i]))),mul(ifac[i],f[n-i])));
    return 0;
}
原文地址:https://www.cnblogs.com/hbyer/p/10715112.html