luogu 【模板】分治 FFT

背景：

题目背景有提示啊。

题目传送门：

https://www.luogu.org/problemnew/show/P4721

题意：

给定一个数组$f$，求$g_i={\sum }_{j=1}^i{g}_{i-j}{f}_j$。

思路$1$：

考虑写出$g$的生成函数。
$$\begin{array}{rl}G(x)& =G(x)F(x)+G(0)\\ G(x)& =G(x)F(x)+1\\ G(x)& =\frac{1}{1-F(x)}\end{array}$$

考虑只有$0$至$n-1$次数项有用，因此有：
$$G(x)\equiv \frac{1}{1-F(x)}(\mathrm{m}\mathrm{o}\mathrm{d}{x}^n)$$

多项式求逆即可。
时间复杂度：$\Theta (n\log n)$。

代码$1$：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
const LL mod=998244353,G=3,inv_G=332748118;
using namespace std;
	int a[1000010],b[1000010],f[1000010],g[1000010];
	int limit,n,l,r[1000010];
int dg(int x,int k)
{
	if(!k) return 1;
	int op=dg(x,k>>1);
	if(k&1) return (LL)op*op%mod*x%mod; else return (LL)op*op%mod;
}
int inv(int x)
{
	return dg(x,mod-2);
}
void init(int n)
{
	limit=1,l=0;
	while(limit<(n<<1))
		limit<<=1,l++;
	for(int i=1;i<limit;i++)
	r[i]=((r[i>>1]>>1)|((i&1)<<(l-1)));
}
void NTT(int *now,int limit,int op)
{
	for(int i=0;i<limit;i++)
		if(i<r[i]) swap(now[i],now[r[i]]);
	for(int mid=1;mid<limit;mid<<=1)
	{
		int wn=dg(op==1?G:inv_G,(mod-1)/(mid<<1));
		for(int j=0;j<limit;j+=(mid<<1))
		{
			int w=1;
			for(int k=0;k<mid;k++,w=((LL)w*wn)%mod)
			{
				int x=now[j+k],y=(LL)w*now[j+k+mid]%mod;
				now[j+k]=(x+y)%mod;
				now[j+k+mid]=(x-y+mod)%mod;
			}
		}
	}
}
void dft(int *f,int n,int limit)
{
	NTT(f,limit,-1);
	int INV=inv(limit);
	for(int i=0;i<n;i++)
		f[i]=(LL)f[i]*INV%mod;
}
void poly_inv(int *f,int *g,int n)
{
	if(n==1)
	{
		g[0]=inv(f[0]);
		return;
	}
	poly_inv(f,g,(n+1)>>1);
	init(n);
	memset(a,0,sizeof(a));
	memset(b,0,sizeof(b));
	for(int i=0;i<n;i++)
		a[i]=f[i],b[i]=g[i];
	NTT(a,limit,1),NTT(b,limit,1);
	for(int i=0;i<limit;i++)
		b[i]=(LL)b[i]*((2ll-(LL)a[i]*b[i]%mod+mod)%mod)%mod;
	dft(b,n,limit);
	for(int i=0;i<n;i++)
		g[i]=b[i];
}
int main()
{
	scanf("%d",&n);
	f[0]=1;
	for(int i=1;i<n;i++)
	{
		scanf("%d",&f[i]);
		f[i]=(-f[i]+mod)%mod;
	}
	poly_inv(f,g,n);
	for(int i=0;i<n;i++)
		printf("%d ",g[i]);
}

思路$2$：

不得不学正统的分治$\text{NTT}$了。
考虑分治解决这件事情。
就是对于当前区间$[l,r]$，我们分成$[l,mid],(mid,r]$来完成。
那么中途我们需要计算$[l,mid]$对$(mid,r]$的贡献，容易发现等号右边是一个卷积的形式，$\text{NTT}$优化即可。
时间复杂度：$\Theta (n{\log }^{2}n)$。

代码$2$：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
const LL mod=998244353,G=3,inv_G=332748118;
using namespace std;
	int a[1000010],b[1000010],f[1000010],g[1000010];
	int limit,n,l,r[1000010];
int dg(int x,int k)
{
	if(!k) return 1;
	int op=dg(x,k>>1);
	if(k&1) return (LL)op*op%mod*x%mod; else return (LL)op*op%mod;
}
int inv(int x)
{
	return dg(x,mod-2);
}
void init(int n)
{
	limit=1,l=0;
	while(limit<(n<<1))
		limit<<=1,l++;
	for(int i=1;i<limit;i++)
		r[i]=((r[i>>1]>>1)|((i&1)<<(l-1)));
}
void NTT(int *now,int limit,int op)
{
	for(int i=0;i<limit;i++)
		if(i<r[i]) swap(now[i],now[r[i]]);
	for(int mid=1;mid<limit;mid<<=1)
	{
		int wn=dg(op==1?G:inv_G,(mod-1)/(mid<<1));
		for(int j=0;j<limit;j+=(mid<<1))
		{
			int w=1;
			for(int k=0;k<mid;k++,w=((LL)w*wn)%mod)
			{
				int x=now[j+k],y=(LL)w*now[j+k+mid]%mod;
				now[j+k]=(x+y)%mod;
				now[j+k+mid]=(x-y+mod)%mod;
			}
		}
	}
}
void dft(int *f,int n,int limit)
{
	NTT(f,limit,-1);
	int INV=inv(limit);
	for(int i=0;i<n;i++)
		f[i]=(LL)f[i]*INV%mod;
}
void poly_CDQ(int *f,int *g,int l,int r)
{
	if(l==r) return;
	int mid=(l+r)>>1;
	poly_CDQ(f,g,l,mid);
	init(r-l+1);
	for(int i=0;i<limit;i++)
		a[i]=b[i]=0;
	for(int i=l;i<=mid;i++)
		a[i-l]=f[i];
	for(int i=0;i<=r-l;i++)
		b[i]=g[i];
	NTT(a,limit,1),NTT(b,limit,1);
	for(int i=0;i<limit;i++)
		a[i]=(LL)a[i]*b[i]%mod;
	dft(a,r-l+1,limit);
	for(int i=mid+1;i<=r;i++)
		f[i]=((LL)f[i]+a[i-l])%mod;
	poly_CDQ(f,g,mid+1,r);
}
int main()
{
	scanf("%d",&n);
	f[0]=1;
	for(int i=1;i<n;i++)
		scanf("%d",&g[i]);
	poly_CDQ(f,g,0,n-1);
	for(int i=0;i<n;i++)
		printf("%d ",f[i]);
}