【LOJ6261】一个人的高三楼（生成函数）（NTT）

传送门

---

题解：

设$F_0(x)$为原序列的生成函数。$F_k(x)$是进行了$k$次求和的序列的生成函数

考虑找到一个多项式$G(x)$，使得$F_{k-1}(x)\cdot G(x)=F_k(x)(\mathrm{m}\mathrm{o}\mathrm{d}{x}^n)$

很显然一个合法的$G(x)=\sum _{i=0}^n{x}^i$

则$F_k(x)=F_0(x)\cdot G^k(x)$

直接多项式快速幂复杂度为$O(n\log n)$应该可以过。

但是我们想要一个更加接近数学本质的做法。

考虑这个多项式$G^k(x)$。我们考虑其第$i$项系数的组合意义，实际上就是一个左上角为$(0,0)$，右下角为$(k,i)$的矩形，从左上角单调走到右下角的方案数，为$\left(\genfrac{}{}{0ex}{}{k+i-1}{i-1}\right)$

直接卷积。

---

代码：

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

using std::cerr;
using std::cout;

cs int mod=998244353,g=3,invg=(mod+1)/g;
inline int add(int a,int b){return (a+=b)>=mod?(a-mod):a;}
inline int dec(int a,int b){return (a-=b)<0?(a+mod):a;}
inline int mul(int a,int b){ll r=(ll)a*b;return r>=mod?r%mod:r;}
inline int power(int a,int b,int res=1){
	for(;b;b>>=1,a=mul(a,a))(b&1)&&(res=mul(res,a));
	return res;
}

cs int N=1<<19|1;
int r[N];

inline void NTT(int *A,int len,int typ){
	if(typ==-1)std::reverse(A+1,A+len);
	for(int re i=0;i<len;++i)if(i<r[i])std::swap(A[i],A[r[i]]);
	for(int re i=1;i<len;i<<=1){
		int wn=power(3,(mod-1)/i/2);
		for(int re j=0;j<len;j+=i<<1)
		for(int re k=0,x,y,w=1;k<i;++k,w=mul(w,wn)){
			x=A[j+k],y=mul(A[j+k+i],w);
			A[j+k]=add(x,y);A[j+k+i]=dec(x,y);
		}
	}
	if(typ==-1)for(int re i=0,inv=power(len,mod-2);i<len;++i)A[i]=mul(A[i],inv);
}
inline void init_rev(int len){
	for(int re i=1;i<len;++i)r[i]=r[i>>1]>>1|((i&1)*(len>>1));
}

int n;ll k;
int F[N],G[N],inv[N];

signed main(){
//	freopen("hitoli.in","r",stdin);
	std::cin>>n>>k;k%=mod;
	inv[0]=inv[1]=G[0]=1;
	for(int re i=0;i<n;++i){
		scanf("%d",F+i);
		(i>1)&&(inv[i]=mul(inv[mod%i],mod-mod/i));
		(i>0)&&(G[i]=mul(G[i-1],mul(i+k-1,inv[i])));
	}
	int len=1;
	while(len<n*2)len<<=1;
	init_rev(len);
	NTT(F,len,1);NTT(G,len,1);
	for(int re i=0;i<len;++i)F[i]=mul(F[i],G[i]);
	NTT(F,len,-1);
	for(int re i=0;i<n;++i)cout<<F[i]<<"\n";
	return 0;
}