[luogu5162]WD与积木

设$g_{n}$表示$n$个积木放的方案数，枚举最后一层所放的积木，则有$g_{n}=sum_{i=1}^{n}c(n,i)g_{n-i}$（因为积木有编号的所以要选出$i$个）

将组合数展开并化简，得到$frac{g_{n}}{n!}=sum_{i=1}^{n}frac{g_{n-i}}{i!(n-i)!}$，明显是一个分治fft的形式，转换为多项式求逆：

令$g_{n}$的指数生成函数为$G(x)=sum_{i=0}^{infty}frac{g_{i}}{i!}x^{i}$，再令$H(x)=sum_{i=0}^{infty}frac{x^{i}}{i!}$，考虑$H(x)G(x)$，代入并将$i=0$和$j=0$提出可得$H(x)G(x)=sum_{i=0}^{infty}(sum_{j=0}^{i}H(x)[j]cdot G(x)[i-j])x^{i}=2G(x)-1$，即$G(x)=frac{1}{2-H(x)}$

设$f_{n}$表示$n$个积木放的方案数的层数之和，那么根据期望的定义，答案$E=frac{f_{n}}{g_{n}}$

类似的，可以写出$f_{n}$的递推式，即$f_{n}=g_{n}+sum_{i=1}^{n}c(n,i)f_{n-i}$（$g_{n}$是因为每一个方案都要+1），将$g_{n}$的表达式展开，有$frac{f_{n}}{n!}=sum_{i=1}^{n}frac{f_{n-i}+g_{n-i}}{i!(n-i)!}$，同样去转换为多项式求逆：

令$f_{n}$的指数生成函数为$F(x)=sum_{i=0}^{infty}frac{f_{i}}{i!}x^{i}$，$H(x)$和$G(x)$相同，考虑$H(x)(G(x)+F(x))$，类似上面可得$H(x)(G(x)+F(x))=G(x)+2F(x)$（注意$f_{0}=0$），即$F(x)=frac{(H(x)-1)G(x)}{2-H(x)}=(H(x)-1)G(x)^{2}$

最终答案即$frac{F(x)[n]}{G(x)[n]}$（都除以了$n!$），注意最后一步不能将$frac{F(x)}{G(x)}$作为结果，因为这不等于答案

#include<bits/stdc++.h>
using namespace std;
#define N (1<<18)
#define mod 998244353
int t,n,a[N],c[N],h[N],g[N],f[N];
int ksm(int n,int m){
    if (!m)return 1;
    int s=ksm(n,m>>1);
    s=1LL*s*s%mod;
    if (m&1)s=1LL*s*n%mod;
    return s;
}
void ntt(int *a,int n,int p){
    for(int i=0;i<(1<<n);i++){
        int rev=0;
        for(int j=0;j<n;j++)rev=rev*2+((i&(1<<j))>0);
        if (i<rev)swap(a[i],a[rev]);
    }
    for(int i=2;i<=(1<<n);i*=2){
        int s=ksm(3,(mod-1)/i);
        if (p)s=ksm(s,mod-2);
        for(int j=0;j<(1<<n);j+=i)
            for(int k=0,ss=1;k<(i>>1);k++,ss=1LL*ss*s%mod){
                int x=a[j+k],y=1LL*a[j+k+(i>>1)]*ss%mod;
                a[j+k]=(x+y)%mod;
                a[j+k+(i>>1)]=(x+mod-y)%mod;
            }
    }
    if (p){
        int s=ksm((1<<n),mod-2);
        for(int i=0;i<(1<<n);i++)a[i]=1LL*a[i]*s%mod;
    }
}
void Inv(int *a,int *b,int n){
    if (!n){
        b[0]=ksm(a[0],mod-2);
        return;
    }
    Inv(a,b,n-1);
    for(int i=0;i<(1<<n);i++)c[i]=a[i];
    for(int i=(1<<n);i<(1<<n+1);i++)c[i]=0;
    ntt(c,n+1,0);
    ntt(b,n+1,0);
    for(int i=0;i<(1<<n+1);i++)b[i]=1LL*b[i]*(mod+2-1LL*c[i]*b[i]%mod)%mod;
    ntt(b,n+1,1);
    for(int i=(1<<n);i<(1<<n+1);i++)b[i]=0;
}
int main(){
    a[0]=1;
    for(int i=1;i<N;i++)a[i]=1LL*a[i-1]*ksm(i,mod-2)%mod;
    h[0]=(mod+2-a[0])%mod;
    for(int i=1;i<N/2;i++)h[i]=(mod-a[i])%mod;
    Inv(h,g,17);
    memcpy(f,g,sizeof(g));
    ntt(f,18,0);
    for(int i=0;i<N;i++)f[i]=1LL*f[i]*f[i]%mod;
    ntt(f,18,1);
    for(int i=N/2;i<N;i++)f[i]=0;
    ntt(f,18,0);
    h[0]=0;
    for(int i=1;i<N/2;i++)h[i]=a[i];
    ntt(h,18,0);
    for(int i=0;i<N;i++)f[i]=1LL*f[i]*h[i]%mod;
    ntt(f,18,1);
    scanf("%d",&t);
    while (t--){
        scanf("%d",&n);
        printf("%lld
",1LL*f[n]*ksm(g[n],mod-2)%mod);
    }
}