P5162 WD与积木
题目背景
WD整日沉浸在积木中，无法自拔……
题目描述
WD想买\(n\)块积木，商场中每块积木的高度都是\(1\)，俯视图为正方形（边长不一定相同）。由于一些特殊原因，商家会给每个积木随机一个大小并标号，发给WD。
接下来WD会把相同大小的积木放在一层，并把所有层从大到小堆起来。WD希望知道所有不同的堆法中层数的期望。两种堆法不同当且仅当某个积木在两种堆法中处于不同的层中，由于WD只关心积木的相对大小，因此所有堆法等概率出现，而不是随机的大小等概率（可以看样例理解）。输出结果\(\bmod 998244353\)即可。
（如果还是不能够理解题意，请看样例）
输入输出格式
输入格式：
第一行一个数\(T\)，表示询问个数。
接下来\(T\)行每行一个数\(n\)，表示WD希望使用\(n\)块积木。
输出格式：
共\(T\)行，每行一个数表示答案\(\bmod 998244353\)。
说明
subtask1(21pts): \(1≤T≤1,000, 1≤n≤1,000\)
subtask2(37pts):\(~1\le T\le 10,~1\le n\le 100,000\)
subtask3(42pts):\(~1\le T\le 100,000,~1\le n\le 100,000\)
当时拿斯特林数推了一会儿没推出来就走了，原来只是部分分啊。
姿势水平屑了，今天稍微给自己普及了一些指数型生成函数。
朴素DP
\(g_n\)代表\(n\)个东西的堆法，就是有标号球和盒子不准空的方案数，就是有序贝尔数，递推的时候枚举第一堆大小。
\[ g_n=\sum_{i=1}^n \binom{n}{i}g_{n-i} \]
\(f_i\)代表\(n\)个东西所有堆法的总层数。
\[ f_n=g_n+\sum_{i=1}^n\binom{n}{i}f_{n-i} \]
从意义上看，\(f_0=0,g_0=1\)，不想从意义上看就稍微推一下。
实质还是枚举第一堆的大小，\(g_n\)算的是枚举的这一堆的贡献和，剩下的是其他堆的贡献。
然后构造指数生成函数
\[ F(x)=\sum_{i=0}^\infty\frac{f_i}{i!}x^i,G(x)=\sum_{i=0}^\infty\frac{g_i}{i!}x^i,H(x)=\sum_{i=0}^\infty\frac{1}{i!}x^i \]
然后稍微注意一下发现上面的下标是从\(1\)开始的，于是直接卷会多一个
\[ 2G=GH+1,2F=FH+G-1 \]
关于＋1-1，今天想到一种理解方式。
\(1\)实际上是一个多项式单位元，像数论卷积定义的\(\epsilon(n)=[n=1]\)那样。
然后＋1还是为了\(g_0=1\)考虑的，因为本身这个东西已经封闭了，你得给它打开一个开口啊（天呐我在扯什么
后面-1减的实际上还是\(g_0=1\)
化简一下式子
\[ F=G(G-1) \]
求逆求出\(G\)就行了
Code:
#include <cstdio>
#include <algorithm>
const int mod=998244353,Gi=332748118;
const int M=(1<<18)+10;
const int N=1e5;
#define mul(a,b) (1ll*(a)*(b)%mod)
#define add(a,b) ((a+b)%mod)
int qp(int d,int k){int f=1;while(k){if(k&1)f=mul(f,d);d=mul(d,d),k>>=1;}return f;}
int fac[M],inv[M],ans[M],H[M],A[M],B[M],D[N],turn[M],len=1;
void NTT(int *a,int typ,int len)
{
int L=-1;for(int i=1;i<len;i<<=1) ++L;
for(int i=1;i<len;i++)
{
turn[i]=turn[i>>1]>>1|(i&1)<<L;
if(i<turn[i]) std::swap(a[i],a[turn[i]]);
}
for(int le=1;le<len;le<<=1)
{
int wn=qp(typ?3:Gi,(mod-1)/(le<<1));
for(int p=0;p<len;p+=le<<1)
{
int w=1;
for(int i=p;i<p+le;i++,w=mul(w,wn))
{
int tx=a[i],ty=mul(w,a[i+le]);
a[i]=add(tx,ty);
a[i+le]=add(tx,mod-ty);
}
}
}
if(!typ)
{
int inv=qp(len,mod-2);
for(int i=0;i<len;i++) a[i]=mul(a[i],inv);
}
}
void polyinv(int *a,int *b,int len)
{
if(len==1){b[0]=qp(a[0],mod-2);return;}
polyinv(a,b,len>>1);
for(int i=0;i<len;i++) A[i]=b[i],B[i]=a[i],A[i+len]=B[i+len]=0;
NTT(A,1,len<<1),NTT(B,1,len<<1);
for(int i=0;i<len<<1;i++) A[i]=mul(A[i],add(2,mod-mul(A[i],B[i])));
NTT(A,0,len<<1);
for(int i=0;i<len;i++) b[i]=A[i];
}
int main()
{
fac[0]=1;for(int i=1;i<=N;i++) fac[i]=mul(fac[i-1],i);
inv[N]=qp(fac[N],mod-2);
for(int i=N-1;~i;i--) inv[i]=mul(inv[i+1],i+1);
for(int i=0;i<=N;i++) H[i]=mod-inv[i];H[0]=1;
while(len<=N) len<<=1;
polyinv(H,ans,len);
for(int i=0;i<len;i++) D[i]=mul(ans[i],fac[i]),H[i]=ans[i];
--ans[0];
NTT(ans,1,len<<1),NTT(H,1,len<<1);
for(int i=0;i<len<<1;i++) ans[i]=mul(ans[i],H[i]);
NTT(ans,0,len<<1);
for(int i=0;i<=N;i++) ans[i]=mul(qp(D[i],mod-2),mul(ans[i],fac[i]));
int T,n;scanf("%d",&T);
while(T--)
{
scanf("%d",&n);
printf("%d\n",ans[n]);
}
return 0;
}2018.12.31