bzoj 2194: 快速傅立叶之二 fft

题意

请计算C[k]=sigma(a[i]*b[i-k]) 其中 k < = i < n ，并且有 n < = 10 ^ 5。 a,b中的元素均为小于等于100的非负整数。

分析

把数组a取反即变成了和为定值的形式，直接上fft即可。
注意精度问题。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<complex>
#define pi acos(-1)
#define N 300005
using namespace std;

typedef complex<double> com;

com a[N],b[N];
int n,rev[N],m,lg;

void bitrev(int n)
{
    for (int i=0;i<n;i++) rev[i]=(rev[i>>1]>>1)|((i&1)<<(lg-1));
}

void fft(com *a,int f,int n)
{
    for (int i=0;i<n;i++)
        if (rev[i]<i) swap(a[i],a[rev[i]]);
    for (int i=1;i<n;i<<=1)
    {
        com wn(cos(pi/i),f*sin(pi/i));
        for (int j=0;j<n;j+=i<<1)
        {
            com w(1,0);
            for (int k=0;k<i;k++)
            {
                com u=a[j+k],v=a[j+k+i]*w;
                a[j+k]=u+v;a[j+k+i]=u-v;
                w*=wn;
            }
        }
    }
    if (f==-1)
    {
        for (int i=0;i<n;i++) a[i]/=n;
    }
}

int main()
{
    scanf("%d",&n);
    for (int i=0;i<n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        a[n-i-1]=x;b[i]=y;
    }
    m=n*2-2;
    for (int i=1;i<=m;i*=2,lg++);
    bitrev(1<<lg);
    fft(a,1,1<<lg);fft(b,1,1<<lg);
    for (int i=0;i<(1<<lg);i++) a[i]*=b[i];
    fft(a,-1,1<<lg);
    for (int i=n-1;i>=0;i--) printf("%d\n",(int)(a[i].real()+0.1));
    return 0;
}