bzoj2194 快速傅立叶之二

最新推荐文章于 2020-09-21 22:30:57 发布

elijahqi

最新推荐文章于 2020-09-21 22:30:57 发布

阅读量195

点赞数

分类专栏： fft 数学

本文链接：https://blog.csdn.net/elijahqi/article/details/79269586

版权

数学同时被 2 个专栏收录

149 篇文章 0 订阅

订阅专栏

fft

7 篇文章 0 订阅

订阅专栏

本文介绍了一个计算卷积C[k]的问题，即求解C[k]=sigma(a[i]*b[i-k])，其中k<=i<n，n<=10^5，a和b中的元素均为小于等于100的非负整数。输入包含整数N及数组a和b的值，输出为计算后的C[k]值。文章提供了使用快速傅里叶变换(FFT)的方法来高效地解决此问题。

摘要由CSDN通过智能技术生成

http://www.elijahqi.win/2018/02/06/bzoj2194/
Description

请计算C[k]=sigma(a[i]*b[i-k]) 其中 k < = i < n ，并且有 n < = 10 ^ 5。 a,b中的元素均为小于等于100的非负整数。
Input
第一行一个整数N,接下来N行，第i+2..i+N-1行，每行两个数，依次表示a[i],b[i] (0 < = i < N)。
Output

输出N行，每行一个整数，第i行输出C[i-1]。
Sample Input
5
3 1
2 4
1 1
2 4
1 4
Sample Output
24
12
10
6
1
http://blog.csdn.net/elijahqi/article/details/79249340
通过观察蒟蒻我在这里推导的这一步过程可以得到我只需要在对b计算dft的时候把我的单位根取反即可
输出的时候直接输出前k项即可

#include<cmath>
#include<cstdio>
#include<complex>
#include<algorithm>
#define N 330000
#define pi acos(-1)
#define C complex<double>
using namespace std;
inline char gc(){
    static char now[1<<16],*S,*T;
    if (T==S){T=(S=now)+fread(now,1,1<<16,stdin);if (T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9') {if (ch=='-') f=-1;ch=gc();}
    while(ch<='9'&&ch>='0') x=x*10+ch-'0',ch=gc();
    return x*f;
}
C a[N],b[N];int n,R[N];
inline void fft(C *x,int f){
    for (int i=0;i<n;++i) if (i<R[i]) swap(x[i],x[R[i]]);
    for (int i=1;i<n;i<<=1){
        C wn(cos(pi/i),f*sin(pi/i));
        for (int j=0;j<n;j+=i<<1){
            C w(1,0),t1,t2;
            for (int k=0;k<i;++k,w*=wn) t1=x[j+k],t2=x[i+k+j]*w,x[k+j]=t1+t2,x[i+k+j]=t1-t2;
        }
    }
}
int main(){
    freopen("bzoj2194.in","r",stdin);
    n=read();for (int i=0;i<n;++i) a[i]=read(),b[i]=read();
    n--;int nn=n<<1,l=0;for (n=1;n<=nn;n<<=1,++l);
    for (int i=0;i<n;++i) R[i]=R[i>>1]>>1|(i&1)<<(l-1);
    fft(a,1);fft(b,-1);for (int i=0;i<n;++i) a[i]*=b[i];fft(a,-1);
    for (int i=0;i<=(nn>>1);++i) printf("%d\n",(int)(a[i].real()/n+0.5));

    return 0;
}