CF438E The Child and Binary Tree（生成函数，NTT）

题目链接：洛谷 CF原网

题目大意：有 $n$ 个互不相同的正整数 $c_i$。问对于每一个 $1\le i\le m$，有多少个不同形态（考虑结构和点权）的二叉树满足每个点权都在 $c$ 中出现过，且点权和为 $i$。答案对 $998244353$ 取模。

$1\le n,m\le 10^5$。

---

首先考虑DP，$f_i$ 表示点权和为 $i$ 的树数。

那么枚举根节点的点权和两棵子树的点权和 $f_k=\sum\limits^n_{i=1}c_i\sum\limits^{k-c_i}_{j=0}f_jf_{k-c_i-j}$。

初始状态 $f_0=1$。因为空树也能作为子树。

这样的复杂度是 $O(nm^2)$，不能过。

考虑 $c$ 的生成函数 $C(x)=\sum x^{c_i}$ 和 $f$ 的生成函数 $F(x)=\sum f_ix^i$。（你问我怎么想到的？我也不知道啊）

那么容易发现原来的式子就是几个函数的卷积。

$F=C\times F\times F+1$（注意 $f_0=1$）

$C\times F^2-F+1=0$

$F=\dfrac{1\pm\sqrt{1-4C}}{2C}$

接下来看看上面该取正还是负。

取正时 $\lim\limits_{x\rightarrow 0}F(x)=+\infty$，不收敛，舍去。

取负时 $\lim\limits_{x\rightarrow 0}F(x)=1$，符合题意。

那么 $F=\dfrac{1-\sqrt{1-4C}}{2C}=\dfrac{2}{1+\sqrt{1-4C}}$。

直接套模板即可。时间复杂度 $O(m\log m)$。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=,mod=;
#define FOR(i,a,b) for(int i=(a);i<=(b);i++)
#define ROF(i,a,b) for(int i=(a);i>=(b);i--)
#define MEM(x,v) memset(x,v,sizeof(x))
inline int read(){
    char ch=getchar();int x=,f=;
    while(ch<'' || ch>'') f|=ch=='-',ch=getchar();
    while(ch>='' && ch<='') x=x*+ch-'',ch=getchar();
    return f?-x:x;
}
int n,m,c[maxn],lim,l,rev[maxn],invtmp[maxn],Binv[maxn],sqrtmp[maxn],Csqrt[maxn],Cinv[maxn];
inline void init(int upr){
    for(lim=,l=;lim<upr;lim<<=,l++);
    FOR(i,,lim-) rev[i]=(rev[i>>]>>)|((i&)<<(l-));
}
inline int add(int a,int b){return a+b<mod?a+b:a+b-mod;}
inline int sub(int a,int b){return a<b?a-b+mod:a-b;}
inline int qpow(int a,int b){
    int ans=;
    for(;b;b>>=,a=1ll*a*a%mod) if(b&) ans=1ll*ans*a%mod;
    return ans;
}
void NTT(int *A,int tp){
    FOR(i,,lim-) if(i<rev[i]) swap(A[i],A[rev[i]]);
    for(int i=;i<lim;i<<=)
        for(int j=,r=i<<,Wn=qpow(,mod-+tp*(mod-)/r);j<lim;j+=r)
            for(int k=,w=;k<i;k++,w=1ll*w*Wn%mod){
                int x=A[j+k],y=1ll*A[i+j+k]*w%mod;
                A[j+k]=add(x,y);A[i+j+k]=sub(x,y);
            }
    if(tp==-) for(int i=,linv=qpow(lim,mod-);i<lim;i++) A[i]=1ll*A[i]*linv%mod;
}
void poly_inv(int *A,int *B,int deg){
    if(deg==) return void(B[]=qpow(A[],mod-));
    poly_inv(A,B,(deg+)>>);
    init(deg<<);
    FOR(i,,deg-) invtmp[i]=A[i];
    FOR(i,deg,lim-) invtmp[i]=;
    NTT(invtmp,);NTT(B,);
    FOR(i,,lim-) B[i]=1ll*sub(,1ll*invtmp[i]*B[i]%mod)*B[i]%mod;
    NTT(B,-);
    FOR(i,deg,lim-) B[i]=;
}
void poly_sqrt(int *A,int *B,int deg){
    if(deg==) return void(B[]=);
    poly_sqrt(A,B,(deg+)>>);
    init(deg<<);
    FOR(i,,lim-) Binv[i]=;
    poly_inv(B,Binv,deg);
    init(deg<<);
    FOR(i,,deg-) sqrtmp[i]=A[i];
    FOR(i,deg,lim-) Binv[i]=sqrtmp[i]=;
    NTT(sqrtmp,);NTT(Binv,);
    FOR(i,,lim-) sqrtmp[i]=1ll*sqrtmp[i]*Binv[i]%mod;
    NTT(sqrtmp,-);
    FOR(i,,deg-) B[i]=499122177ll*add(B[i],sqrtmp[i])%mod;
    FOR(i,deg,lim-) B[i]=;
}
int main(){
    n=read();m=read();
    FOR(i,,n){
        int x=read();
        if(x<=m) c[x]=;
    }
    FOR(i,,m) c[i]=(mod-4ll*c[i]%mod)%mod;
    c[]=;
    poly_sqrt(c,Csqrt,m+);
    Csqrt[]=add(Csqrt[],);
    poly_inv(Csqrt,Cinv,m+);
    FOR(i,,m) printf("%d\n",add(Cinv[i],Cinv[i]));
}