【bzoj3527】[Zjoi2014]力 FFT

Description

给出n个数qi，给出Fj的定义如下：

令Ei=Fi/qi，求Ei.
Input

第一行一个整数n。
接下来n行每行输入一个数，第i行表示qi。
n≤100000，0< qi< 1000000000
Output

n行，第i行输出Ei。与标准答案误差不超过1e-2即可。

Sample Input

5

4006373.885184

15375036.435759

1717456.469144

8514941.004912

1410681.345880
Sample Output

-16838672.693

3439.793

7509018.566

4595686.886

10903040.872

题解
http://blog.csdn.net/clover_hxy/article/details/56486609

代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<complex>
#define ll long long
#define mod 10000007
#define E complex<double>
#define pi acos(-1) 
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
const int N=262144;
int n,m,L;
int R[N];
E f[N],f_[N],g[N];
void FFT(E *a,int f)
{
    for (int i=0;i<n;i++) if (i<R[i]) swap(a[i],a[R[i]]);
    for (int i=1;i<n;i<<=1)
    {
        E wn(cos(pi/i),f*sin(pi/i));
        for (int j=0;j<n;j+=(i<<1))
        {
            E w(1,0),x,y;
            for (int k=0;k<i;k++,w*=wn)
            {
                x=a[j+k],y=a[j+i+k]*w;
                a[j+k]=x+y;a[j+i+k]=x-y;
            }
        }
    }
    if (f==-1)for (int i=0;i<n;i++) a[i]/=n;
}
int main()
{
    n=read();n--;
    for (int i=0;i<=n;i++)
    {
        double x;scanf("%lf",&x);
        f[i]=f_[n-i]=x;
    }
    for (int i=1;i<=n;i++) g[i]=1.0/i/i;
    m=2*n;for (n=1;n<=m;n<<=1) L++;
    for (int i=0;i<n;i++) R[i]=(R[i>>1]>>1)|((i&1)<<(L-1));
    FFT(f,1);FFT(g,1);FFT(f_,1);
    for (int i=0;i<n;i++)f[i]=f[i]*g[i];
    for (int i=0;i<n;i++)f_[i]=f_[i]*g[i];
    FFT(f,-1);FFT(f_,-1);
    for (int i=0;i<=m/2;i++)
        printf("%.3lf\n",f[i].real()-f_[m/2-i].real());
    return 0;
}