1. 程式人生 > >P4721【模板】分治 FFT

P4721【模板】分治 FFT

這樣的 turn long long tdi mat memset eof 發現 algorithm

瞎扯

雖然說是FFT但是還是寫了一發NTT(笑)
然後忘了IDFT之後要除個n懵逼了好久
以及遞歸的時候忘了邊界無限RE

思路

樸素算法

分治FFT
考慮到題目要求求這樣的一個式子
\[ F_x=\Sigma_{i=1}^{x}F_{x-i}G_{i} \]

我們可以按定義暴力,然後再松式卡常(不是)
我們可以發現它長得像一個卷積一樣,但是因為後面的f值會依賴與前面的f值,所以沒法一遍FFT直接求出結果,而對每個f都跑一遍FFT太慢了,我們使用分治優化這個過程就很優秀了,復雜度是\(O(n\log^2 n)\)

分治優化

我們能夠想到cdq分治的思想,在統計一個區間時,確保對這個區間有影響的操作產生的貢獻已經全被統計,就是先統計[l,mid]區間對[mid+1,r]區間的貢獻

然後發現對於每個\(f_x\),它對後面的\(f_i\)產生的貢獻是\(\Sigma_{j=l}^{mid} f_{i}g_{i-j}\)

然後分治就好

代碼

#include <cstdio>
#include <algorithm>
#include <cstring>
#define int long long
using namespace std;

const int MOD = 998244353,G=3,invG=332748118;
int a[200000],b[200000],f[200000],g[200000],n;
int pow(int a,int b){
    int ans=1;
    while(b){
        if(b&1)
            ans=(1LL*ans*a)%MOD;
        a=(1LL*a*a)%MOD;
        b>>=1;
    }
    return ans;
}
void FFT(int *a,int opt,int n){
    int lim=0;
    while((1<<lim)<n)
        lim++;
    for(int i=0;i<n;i++){
        int t=0;
        for(int j=0;j<lim;j++)
            if((i>>j)&1)
                t|=(1<<(lim-j-1));
        if(i<t)
            swap(a[i],a[t]);        
    }
    for(int i=2;i<=n;i<<=1){
        int len=i/2;
        int tmp=pow((opt)?G:invG,(MOD-1)/i);
        for(int j=0;j<n;j+=i){
            int arr=1;
            for(int k=j;k<len+j;k++){
                int t=arr*a[k+len];
                a[k+len]=((a[k]-t)%MOD+MOD)%MOD;
                a[k]=(a[k]+t)%MOD;
                arr=(arr*tmp)%MOD;         
            }
        }
    }
    if(opt==0){
        int invt=pow(n,MOD-2);
        for(int i=0;i<n;i++)
            a[i]=a[i]*invt%MOD;
    }
}
void solve(int l,int r){
    if(r-l==1)
        return;
    int t=pow(r-l,MOD-2);
    int mid=(l+r)>>1;
    solve(l,mid);
    memset(a+(r-l)/2,0,sizeof(int)*(r-l)/2);
    memcpy(a,f+l,sizeof(int)*(r-l)/2);
    memcpy(b,g,sizeof(int)*(r-l));
    FFT(a,1,r-l);
    FFT(b,1,r-l);
    for(int i=0;i<r-l;i++)
        a[i]=(a[i]*b[i])%MOD;
    FFT(a,0,r-l);
    for(int i=(r-l)/2;i<r-l;i++)
        f[l+i]=(f[l+i]+a[i])%MOD;
    solve(mid,r);
}
signed main(){
    int mid;
    scanf("%lld",&n);
    mid=n;
    for(int i=1;i<=n-1;i++)
        scanf("%lld",&g[i]);
    int t=1;
    while(t<n)
        t<<=1;
    n=t;
    f[0]=1;
    solve(0,n);
    for(int i=0;i<mid;i++)
        printf("%lld ",f[i]);
    return 0;
}

P4721【模板】分治 FFT