P4721 【模板】分治 FFT（FFT，分治板子理解）

P4721 【模板】分治 FFT

题意：

• 给定序列 $g_{1\dots n-1}$，求序列 $f_{0\dots n-1}$

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

分析：

• 若直接按照题目给的式子展开求和，$O(n^2)$ ，会超时

• 考虑分治：

  先将f_i展开来观察一下：

  $f_1=g_1{f}_0$

  $f_2=g_1{f}_1+g_2{f}_0$

  $f_3=g_1{f}_2+g_2{f}_1+g_3{f}_0$

  $f_4=g_1{f}_3+g_2{f}_2+g_3{f}_1+g_4{f}_0$

  $f_5=g_1{f}_4+g_2{f}_3+g_3{f}_2+g_4{f}_1+g_5{f}_0$

  考虑分治的话，要考虑区间 $[l,r]$ 中，$[l,mid]$ 和 $[mid+1,r]$ 之间的关系

  容易发现，$[l,mid]$ 会对 $[mid+1,r]$ 产生贡献（令其为 $va{l}_i$ ）

  $va{l}_i={\sum }_{j=l}^{mid}{g}_{i-j}{f}_j$ , $i\in [mid+1,r]$

  $va{l}_i$ 是由两个函数卷积得出的，将 $g_{i-j}{f}_j$ 中的 $(i-j),j$ 看成多项式的项数（次数），则其系数为 $f_j$ ，可通过 $NTT$ 加速算出

• 分治过程中区间之间的关系：

  • 令 $l=0$ ，$r=7$
  • $(l,mid)-->(mid+1,r)$
    1. $(0,0)-->(1,1)$
    2. $(0,1)-->(2,3)$
    3. $(2,2)-->(3,3)$
    4. $(0,3)-->(4,7)$
    5. $(4,4)-->(5,5)$
    6. $(4,5)-->(6,7)$
    7. $(6,6)-->(7,7)$
  • $f_i=$过程中算得的 $va{l}_i$ 之和
    时间复杂度：$O(nlo{g}^{2}n)$ , 分治一个 $log$

#include<bits/stdc++.h>
#define int long long 
using namespace std;

const int N = (1<<20)+5, mo=998244353;

inline int binpow(int a,int b)
{
    int ans=1;
    while(b)
    {
        if(b&1) ans = ans*a%mo;
        a = a*a%mo;
        b >>= 1;
    }
    return ans;
}
int rev[N];
void ntt(int *a, int n, int inv)
{
	for(int i=0;i<n;i++)
    {
		if(i < rev[i]) swap(a[i], a[rev[i]]);
    }
	for(int len=1;len<n;len<<=1) 
    {	
		int Wn = binpow(3, (mo-1)/(len<<1));
        if(inv == -1) Wn = binpow(Wn, mo-2);
		for(int i=0;i<n;i+=(len<< 1)) 
        {
			int w=1;
			for(int j=0;j<len;j++, w = (w*Wn)%mo) 
            {
				int x = a[i + j], y = w*a[i+j+len]%mo;
				a[i+j] = (x+y)%mo; a[i+j+len] = (x-y+mo)%mo;
			}
		}
	}
    if(inv == -1)
    {
        int fg=binpow(n, mo-2);
        for(int i=0;i<n;i++) a[i] = a[i]*fg%mo;
    }
}
inline void mul(int *a,int *b,int k)
{
    int s=1<<k;
    for(int i=1;i<s;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(k-1));
    ntt(a,s,1); ntt(b,s,1);
    for(int i=0;i<s;i++) a[i] = a[i]*b[i]%mo;
    ntt(a,s,-1);
}
int g[N], a[N], b[N], f[N];
inline void binary(int l,int r)
{
    if(r == l) return ;
    int mid = (l+r)>>1;
    binary(l, mid);
    int sum=2, k=1;
    while(sum <= (mid-l+r-l)) k++, sum <<= 1;
    for(int i=0;i<sum;i++) a[i] = b[i] = 0;
    for(int i=l;i<=mid;i++) a[i-l] = f[i];
    for(int i=1;i<=r-l;i++) b[i] = g[i];
    mul(a, b, k);
    for(int i=mid+1;i<=r;i++) f[i] = (f[i]+a[i-l])%mo;
    binary(mid+1, r);
}
signed main()
{
    ios::sync_with_stdio(false);cin.tie(nullptr);cout.tie(nullptr);
    int n;
    cin>>n;
    f[0] = 1;
    for(int i=1;i<n;i++) cin>>g[i];
    binary(0, n-1);
    for(int i=0;i<n;i++) cout<<f[i]<<' ';

	return 0;
}