BZOJ 2194 快速傅里叶之二

时间：2021-03-22 20:28:37

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 ;

}

相关文章