[bzoj2194][快速傅里叶]快速傅立叶之二

2194: 快速傅立叶之二

Time Limit: 10 Sec Memory Limit: 259 MB
Submit: 1686 Solved: 996
[Submit][Status][Discuss]
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
HINT

Source

sol：

fft裸题

#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
using namespace std;
typedef double s64;
const s64 pi=acos(-1);
int n,m,g;
inline int read()
{
    char c;
    int res,flag=0;
    while((c=getchar())>'9'||c<'0') if(c=='-')flag=1;
    res=c-'0';
    while((c=getchar())>='0'&&c<='9') res=(res<<3)+(res<<1)+c-'0';
    return flag?-res:res;
}
const int N=510000;
int nx,lg,rev[N];
struct cc
{
    s64 r,i;
    friend inline cc operator +(const cc &a,const cc &b)
    {
        return (cc){a.r+b.r,a.i+b.i};
    }
    friend inline cc operator -(const cc &a,const cc &b)
    {
        return (cc){a.r-b.r,a.i-b.i};
    }
    friend inline cc operator *(const cc &a,const cc &b)
    {
        return (cc){a.r*b.r-a.i*b.i,a.r*b.i+a.i*b.r};
    }
}a[N],b[N];
inline void SET(int n)
{
    nx=1;
    while(nx<n) nx<<=1,++lg;
    for(int i=1;i<nx;++i)
    rev[i]=(rev[i>>1]>>1)|((i&1)<<lg-1);
}
inline void DFT(cc *c,int o)
{
    for(int i=0;i<nx;++i)
    if(rev[i]>i) swap(c[i],c[rev[i]]);

    int len;
    cc w,wn,l,r; 
    for(int k=1;k<nx;k<<=1)
    {
        len=k<<1;
        wn=(cc){cos(pi/k),o*sin(pi/k)};
        for(int s=0;s<nx;s+=len)
        {
            w=(cc){1,0};
            for(int i=s;i<s+k;++i)
            {
                l=c[i];
                r=w*c[i+k];
                c[i]=l+r;
                c[i+k]=l-r;
                w=w*wn;
            }
        }
    }
}
inline void Mul(cc *a,cc *b)
{
    DFT(a,1);DFT(b,1);
    for(int i=0;i<nx;++i) a[i]=a[i]*b[i];
    DFT(a,-1);
    for(int i=0;i<nx;++i) a[i].r/=nx;
}
int main()
{
//  freopen("2194.in","r",stdin);
//  freopen(".out","w",stdout);
    n=read()-1;
    for(int i=0;i<=n;++i)
    {
        a[n-i].r=read();
        b[i].r=read();
    }
    SET(n+n+2);
    Mul(a,b);
    for(int i=n;i>=0;--i) printf("%d\n",(int)(a[i].r+0.5));
}