BZOJ 2194 快速傅里叶之二

fft。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<complex>
#define maxn 300500
#define pi acos(-1)
using namespace std;
typedef complex<double> E;
int n,m,l=,c[maxn],r[maxn];
E a[maxn],b[maxn];
void fft(E *x,int f)
{
    for (int i=;i<n;i++)
        if (i<r[i]) swap(x[i],x[r[i]]);
    for (int i=;i<n;i<<=)
    {
        E wn(cos(pi/i),f*sin(pi/i));
        for (int j=;j<n;j+=(i<<))
        {
            E w(,);
            for (int k=;k<i;k++)
            {
                E r1,r2;
                r1=x[j+k];r2=w*x[i+j+k];
                x[j+k]=r1+r2;x[j+k+i]=r1-r2;
                w*=wn;
            }
        }
    }
    if (f==-)
    {
        for (int i=;i<n;i++)
            x[i]/=n;
    }
}
int main()
{
    scanf("%d",&n);n--;
    for (int i=;i<=n;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        a[i].real()=x;b[n-i].real()=y;
    }
    m=*n;
    for (n=;n<=m;n<<=) l++;
    for (int i=;i<n;i++) r[i]=(r[i>>]>>)|((i&)<<(l-));
    fft(a,);fft(b,);
    for (int i=;i<n;i++) a[i]*=b[i];
    fft(a,-);
    for(int i=m/;i<=m;i++)
        printf("%d\n",(int)(a[i].real()+0.1));
    return ;
}