P4721【模板】分治 FFT
阿新 • • 發佈:2019-02-25
這樣的 turn long long tdi mat memset eof 發現 algorithm
然後發現對於每個\(f_x\),它對後面的\(f_i\)產生的貢獻是\(\Sigma_{j=l}^{mid} f_{i}g_{i-j}\)
瞎扯
雖然說是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]區間的貢獻
然後分治就好
代碼
#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