分治FFT（NTT）

前置知识

• $cdq$分治
• 快速傅里叶变换（$FFT$）
• 快速数论变换（$NTT$）

前言

分治$FFT$是基于分治的算法，通过每次计算左区间对右区间的贡献，来降低$FFT$的时间复杂度。

情景代入

洛谷P4721 【模板】分治 FFT

给定序列$g_1,g_2\dots ,g_{n-1}$，求$f_0,f_1,\dots ,f_{n-1}$。

其中$f_i=\sum _{j=1}^i{f}_{i-j}{g}_j$，边界为$f_0=1$。

答案对$998244353$取模。

$2\le n\le 10^5,0\le g_i<998244353$

分析

由$f$的递推式可看出，它是其前面的项于多项式$g$的卷积。多项式$f$不能一次全部求出，因为其每一项都和前面的项有关。我们可以用分治，将左半区间对右半区间的贡献提前累加到右半区间。

设当前左半区间为$[l,mid]$，右半区间为$[mid+1,r]$，$k\in [mid+1,r]$，考虑左半区间对右半区间内的点$k$的贡献。

$v_k=\sum _{i=l}^{mid}{f}_i\times g_{k-i}$

设$F(x)=\sum _{i=0}^{mid-l}{f}_{i+mid}{x}_i,G(x)=\sum _{i=0}^{r-l}{g}_i{x}^i$，$H(x)=F\ast G$，则

$v_k=H_{k-l}$

所以左半区间$[l,mid]$对右半区间$[mid+1,r]$的贡献可以通过一次多项式乘法得出。

时间复杂度为$O(n{\log }^{2}n)$。

code

#include<bits/stdc++.h>
using namespace std;
long long g[500005],f[500005],a1[500005],a2[500005];
const long long G=3,mod=998244353;
long long mi(long long t,long long v){
	if(!v) return 1;
	long long re=mi(t,v/2);
	re=re*re%mod;
	if(v&1) re=re*t%mod;
	return re;
}
void ch(long long *a,int l){
	for(int i=1,j=l/2,k;i<l-1;i++){
		if(i<j) swap(a[i],a[j]);
		k=l/2;
		while(j>=k){
			j-=k;k>>=1;
		}
		j+=k;
	}
}
void ntt(long long *a,int l,int fl){
	long long wn,w;
	for(int i=2;i<=l;i<<=1){
		if(fl==1) wn=mi(G,(mod-1)/i);
		else wn=mi(G,mod-1-(mod-1)/i);
		for(int j=0;j<l;j+=i){
			w=1;
			for(int k=j;k<j+i/2;k++,w=w*wn%mod){
				long long t=a[k],u=w*a[k+i/2]%mod;
				a[k]=(t+u)%mod;
				a[k+i/2]=(t-u+mod)%mod;
			}
		}
	}
	if(fl==-1){
		long long ny=mi(l,mod-2);
		for(int i=0;i<l;i++) a[i]=a[i]*ny%mod;
	}
}
void solve(int l,int r){
	if(l==r){
		f[l]=(f[l]+g[l])%mod;
		return;
	}
	int mid=l+r>>1;
	solve(l,mid);
	int len=1;
	while(len<r-l+1) len<<=1;
	
	for(int i=0;i<len;i++) a1[i]=a2[i]=0;
	for(int i=0;i<=mid-l;i++) a1[i]=f[i+l];
	for(int i=0;i<=r-l;i++) a2[i]=g[i];
	
	ch(a1,len);ch(a2,len);
	ntt(a1,len,1);ntt(a2,len,1);
	
	for(int i=0;i<len;i++){
		a1[i]=a1[i]*a2[i]%mod;
	}
	
	ch(a1,len);
	ntt(a1,len,-1);
	for(int i=mid+1;i<=r;i++){
		f[i]=(f[i]+a1[i-l])%mod;
	}
	solve(mid+1,r);
}
int main()
{
	int n;
	scanf("%d",&n);
	for(int i=1;i<n;i++){
		scanf("%lld",&g[i]);
	}
	solve(0,n-1);
	f[0]=1;
	for(int i=0;i<=n-1;i++){
		printf("%lld ",f[i]);
	}
	return 0;
}