解题：洛谷4721 [模板]分治FFT

题面

这是CDQ入门题，不要被题目名骗了，这核心根本不在不在FFT上啊=。=

因为后面的项的计算依赖于前面的项，不能直接FFT。所以用CDQ的思想，算出前面然后考虑给后面的贡献

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=100005,mod=998244353;
int a[4*N],b[4*N],rev[4*N],f[N],g[N],n,G,Gi;
void exGCD(int a,int b,int &x,int &y)
{
    if(!b) {x=1,y=0; return ;}
    exGCD(b,a%b,y,x),y-=a/b*x;
}
int Qpow(int x,int k)
{
    if(k==1) return x;
    int tmp=Qpow(x,k/2);
    return k%2?1ll*tmp*tmp%mod*x%mod:1ll*tmp*tmp%mod;
}
int Inv(int x,int m)
{
    int xx,yy;
    exGCD(x,m,xx,yy);
    return (xx%m+m)%m;
}
void NTT(int *arr,int len,int typ)
{
    for(int i=0;i<=len;i++)
        if(rev[i]>i) swap(arr[rev[i]],arr[i]);
    for(int i=2;i<=len;i<<=1)
    {
        int lth=i>>1,ort=Qpow(~typ?G:Gi,(mod-1)/i);
        for(int j=0;j<len;j+=i)
        {
            int ori=1,tmp;
            for(int k=j;k<j+lth;k++,ori=1ll*ori*ort%mod)
            {
                tmp=1ll*ori*arr[k+lth]%mod;
                arr[k+lth]=(arr[k]-tmp+mod)%mod;
                arr[k]=(arr[k]+tmp)%mod;
            }
        }
    }
    if(typ==-1)
        for(int i=0,ni=Inv(len,mod);i<len;i++) 
            arr[i]=1ll*arr[i]*ni%mod;
}
void CDQ(int l,int r,int mid)
{
    int len=r-l+1,m=1;
    for(int i=l;i<=mid;i++) a[i-l]=f[i];
    for(int i=0;i<len;i++) b[i]=g[i]; len+=mid-l+1;
    while(m<=len) m<<=1;
    for(int i=1;i<=m;i++) rev[i]=(rev[i>>1]>>1)+(i&1)*(m>>1);
    NTT(a,m,1),NTT(b,m,1);
    for(int i=0;i<=m;i++) a[i]=1ll*a[i]*b[i]%mod;
    NTT(a,m,-1);
    for(int i=mid+1;i<=r;i++) f[i]+=a[i-l],f[i]%=mod;
    for(int i=0;i<=m;i++) a[i]=b[i]=0;
}
void Divide(int l,int r)
{
    if(l==r) return;
    int mid=(l+r)/2;
    Divide(l,mid),CDQ(l,r,mid),Divide(mid+1,r);
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++) scanf("%d",&g[i]);
    f[0]=1,G=3,Gi=Inv(G,mod),Divide(0,n-1);
    for(int i=0;i<n;i++) printf("%d ",f[i]);
    return 0;
}