FFT模版題。

觀察題目，我們可以發現，只要把序列b倒過來，再聯想一下乘法運算。。。

我們會發現，將序列a和序列b當作100進制數，做一次乘法，然后從低到高每一位便是答案了（乘完無需進位）

#include <cstdio> #include <cstdlib> #include <cstring> #include <cmath> #include <algorithm> #include <iostream> #include <cctype> #include <complex> #define rep(i, l, r) for(int i=l; i<=r; i++) #define down(i, l, r) for(int i=l; i>=r; i--) #define maxn 400009 #define cd complex <double> #define PI acos(0.0)*2.0 #define ll long long using namespace std; inline int read() {int x=0, f=1; char ch=getchar();while (!isdigit(ch)) {if (ch=='-') f=-1; ch=getchar();}while (isdigit(ch)) x=x*10+ch-'0', ch=getchar();return x*f; } cd a[maxn], b[maxn], c[maxn], A[maxn]; int n, m, n1[maxn], n2[maxn], s[maxn]; int ans[maxn];void fft(cd *a, bool flag) {rep(i, 0, n-1) s[i]=0;for(int i=1, j=n; i<n; i*=2, j/=2) rep(h, j/2, j-1) s[h]+=i;for(int i=1; i<n; i*=2) rep(j, 0, i-1) s[j+i]+=s[j];rep(i, 0, n-1) A[i]=a[s[i]];double pi=flag?PI:-PI;for(int step=1; step<n; step*=2){cd e=exp(cd(0, 2.0*pi/double(step*2))), w=cd(1, 0);for (int pos=0; pos<step; pos++, w*=e) for(int i=pos; i<n; i+=step*2){cd ret=A[i], rec=w*A[i+step];A[i]=ret+rec, A[i+step]=ret-rec;}}if (!flag) rep(i, 0, n-1) A[i]/=n;rep(i, 0, n-1) a[i]=A[i]; }int main() {m=read(); rep(i, 1, m) n1[m-i]=read(), n2[i-1]=read();n=1; while (n<m*2) n*=2;rep(i, 0, n-1) a[i]=cd(n1[i], 0); fft(a, true);rep(i, 0, n-1) b[i]=cd(n2[i], 0); fft(b, true);rep(i, 0, n-1) c[i]=a[i]*b[i]; fft(c, false);rep(i, 0, m-1) ans[i]=c[i].real()+0.5;down(i, m-1, 0) printf("%d\n", ans[i]); }

轉載于:https://www.cnblogs.com/NanoApe/p/4479405.html

總結

以上是生活随笔為你收集整理的BZOJ-2194 快速傅立叶之二的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。