DFT :
1 #include <cstdio> 2 #include <iostream> 3 #include <cmath> 4 #include <complex> 5 typedef double db; 6 typedef long long ll; 7 8 #define com complex<db> 9 using namespace std; 10 const int N=1e5+10; 11 const db pi=acos(-1); 12 int n,h[N*6],M; 13 com q[N*6],r[N*6]; 14 com a[N*6]; 15 void dft(com *src,int sig) { 16 for (int i=0; i<M; i++) a[h[i]]=src[i]; //蝶形变换的准备 17 for (int m=2; m<=M; m<<=1) { //正在求的组大小 18 int half=m>>1; 19 for (int i=0; i<half; i++) { //求A[i]与A[i+k],先枚举这个方便处理主根 20 com w=com(cos(i*2*pi/m) , sig * sin(i*2*pi/m)); 21 //必须一步一求,不然精度会出锅。 22 //由于转移到m,因此按照式子是m次根。 23 for (int j=i; j<M; j+=m) { //第几组 24 int k=j+half; 25 com u=a[j], v=a[k]*w; 26 a[j]=u+v,a[k]=u-v; 27 } 28 } 29 } 30 for (int i=0; i<M; i++) src[i]=a[i]; 31 } 32 int main() { 33 //freopen("3617.in","r",stdin); 34 cin>>n; 35 for (int i=0; i<n; i++) scanf("%lf",&q[i].real()); 36 for (M=1; M<3*n; M<<=1); 37 for (int i=0; i<2*n-1; i++) { 38 if (i==n-1) r[i]=0; else 39 r[i]=pow(i-(n-1),-2) * (i<n-1?-1:1); 40 } 41 for (int i=0; i<M; i++) h[i]=(h[i>>1]>>1) + ((i&1) * (M>>1)); 42 //以次数界-1为长度,翻转二进制 43 44 dft(q,1); dft(r,1); 45 for (int i=0; i<M; i++) q[i]=q[i]*r[i]; 46 dft(q,-1); 47 for (int i=n-1; i<n+n-1; i++) printf("%lf\n",q[i].real() / M); 48 //不要忘记除次数界!!! 49 }
NFT:
#include <cstdio> #include <iostream> using namespace std; const int N=4e5+10,mo=998244353,g=3; typedef long long ll; int n,m,M; int A[N],B[N],h[N]; ll w[N], iw[N]; ll ksm(ll x,ll y) { if (y==0) return 1; if (y==1) return x; ll t=ksm(x,y>>1); return t*t%mo*ksm(x,y&1)%mo; } void ntt(int *a,int sz,int sig) { for (int i = 1; i < sz; i++) h[i] = (h[i>>1]>>1) + (i & 1) * (sz >> 1); for (int i = 0; i <sz; i++) if (h[i]<i) swap(a[i],a[h[i]]); for (int m = 1; m < sz; m<<=1) { int td = M / (m<<1); for (int i = 0; i < sz; i += (m<<1)) { for (int j = 0; j < m; j++) { ll T = a[i+j+m] * (sig == 1 ? w[td * j] : iw[td * j]) % mo; a[i+j+m] = (a[i+j] - T) % mo; a[i+j] = (a[i+j] + T) % mo; } } } } int main() { freopen("test.in","r",stdin); cin>>n>>m;; for (int i=0; i<=n; i++) scanf("%d",&A[i]); for (int i=0; i<=m; i++) scanf("%d",&B[i]); for (M=1; M<=n+m; M<<=1); for (int i=1; i<M; i++) h[i]=(h[i>>1]>>1) + (i&1) * (M>>1); ll ww = ksm(3, (mo - 1) / M); iw[0] = w[0] = 1; for (int i = 1; i < M; i++) w[i] = w[i-1] * ww % mo; ww = ksm(ww, mo - 2); for (int i = 1; i < M; i++) iw[i] = iw[i-1] * ww % mo; ntt(A,M,1); ntt(B,M,1); for (int i=0; i<M; i++) A[i]=(ll)A[i]*B[i]%mo; ntt(A,M,-1); ll cs=ksm(M,mo-2); for (int i=0; i<=n+m; i++) printf("%lld ",(A[i]*cs%mo+mo)%mo); }
FFT:
#include<cstdio> #include<cstring> #include<cmath> #include<algorithm> using namespace std; const int maxn=270010; const double pi=acos(-1.0),eps=1e-4; struct Complex{ double a,b; Complex(double a=0.0,double b=0.0):a(a),b(b){} Complex operator+(const Complex &x)const{return Complex(a+x.a,b+x.b);} Complex operator-(const Complex &x)const{return Complex(a-x.a,b-x.b);} Complex operator*(const Complex &x)const{return Complex(a*x.a-b*x.b,a*x.b+b*x.a);} }A[maxn],B[maxn]; void FFT(Complex*,int,int); int n,m,N=1; int main(){ scanf("%d%d",&n,&m); n++;m++; while(N<n+m)N<<=1; for(int i=0;i<n;i++)scanf("%lf",&A[i].a); for(int i=0;i<m;i++)scanf("%lf",&B[i].a); FFT(A,N,1); FFT(B,N,1); for(int i=0;i<N;i++)A[i]=A[i]*B[i]; FFT(A,N,-1); for(int i=0;i<n+m-1;i++)printf("%d ",(int)(A[i].a+eps)); return 0; } void FFT(Complex *A,int n,int tp){ for(int i=1,j=0;i<n-1;i++){ int k=N; do{ k>>=1; j^=k; }while(j<k); if(i<j)swap(A[i],A[j]); } for(int k=2;k<=n;k<<=1){ Complex wn(cos(-tp*2*pi/k),sin(-tp*2*pi/k)); for(int i=0;i<n;i+=k){ Complex w(1.0,0.0); for(int j=0;j<(k>>1);j++,w=w*wn){ Complex a(A[i+j]),b(w*A[i+j+(k>>1)]); A[i+j]=a+b; A[i+j+(k>>1)]=a-b; } } } if(tp<0)for(int i=0;i<n;i++)A[i].a/=n; }