[P4721] 分治 FFT

「题意」给定(g[0]=1),(g[1~n-1])求序列(f[i]=sum_{j=1}^i f[i-j]*g[j] , iin[1,n-1],f[0]=1)。
「分析」分治处理区间[l,r]，先递归求出[l,mid]，在统计[l,mid]对[mid+1,r]的贡献，可以发现

[f[x]+=sum_{i=l}^{mid}f[i]*g[x-i] , xin[mid+1,r] ]

拿卷积算，令(A[0,mid-l])=(f[l,mid]), (B[l,r-l])=(g[1,r-l]) ,(B[0]=0)，设(C[i])=(A[i]*B[i-j]), 那么

[f[x]+=sum f[i]*g[x-1]=sum A[i-l]*B[x-i]=C[x-l] ]

套上ntt

「代码」

#include <bits/stdc++.h>
using namespace std;
const int N=4e5+10;
const int P=998244353,G=3;

int n,lmt,l,rev[N];
int g[N],f[N],A[N],B[N];

int qpow(int x,int y) {
    int c=1;
    for(; y; y>>=1,x=1LL*x*x%P)
        if(y&1) c=1LL*c*x%P;
    return c;
}
void init(int len) {
    for(lmt=1,l=0; lmt<len+len; lmt<<=1) l++;
    for(int i=0; i<lmt; ++i) rev[i]=(rev[i>>1]>>1)|((i&1)<<(l-1));
}
void numberTheoreticTransform(int a[N],int tp) {
    for(int i=0; i<lmt; ++i) if(i<rev[i]) swap(a[i],a[rev[i]]);
    for(int m=1; m<lmt; m<<=1) {
        long long wm=qpow(G,(P-1)/(m<<1));
        if(tp==-1) wm=qpow(wm,P-2);
        for(int i=0; i<lmt; i+=(m<<1)) {
            long long w=1,tmp;
            for(int j=0; j<m; ++j,w=w*wm%P) {
                tmp=w*a[i+j+m]%P;
                a[i+j+m]=(a[i+j]-tmp+P)%P;
                a[i+j]=(a[i+j]+tmp)%P;
            }
        }
    }
    if(tp==-1) {
        long long tmp=qpow(lmt,P-2);
        for(int i=0; i<lmt; ++i) a[i]=tmp*a[i]%P;
    }
}
void dfs(int l,int r) {
    if(l==r) return;
    int mid=(l+r)>>1;
    dfs(l,mid);
    init(r-l+1);
    for(int i=0; i<lmt; ++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];
    numberTheoreticTransform(A,1);
    numberTheoreticTransform(B,1);
    for(int i=0; i<lmt; ++i) A[i]=1LL*A[i]*B[i]%P;
    numberTheoreticTransform(A,-1);
    for(int i=mid+1; i<=r; ++i) f[i]=(f[i]+A[i-l])%P;
    dfs(mid+1,r);
}


int main() {
    scanf("%d",&n);
    for(int i=1; i<n; ++i) scanf("%d",g+i);
    f[0]=1, dfs(0,n-1);
    for(int i=0; i<n; ++i) printf("%d ",f[i]);
    printf("
");
    return 0;
}

ubuntu的中括号怎么是这个鬼玩意儿//