【ZJOI2014】bzoj3527 力

最新推荐文章于 2018-02-12 21:59:44 发布

sdfzyhx

最新推荐文章于 2018-02-12 21:59:44 发布

阅读量290

点赞数

分类专栏：数学 bzoj 省选真题文章标签： FFT

本文链接：https://blog.csdn.net/sdfzyhx/article/details/54773309

版权

bzoj 同时被 3 个专栏收录

251 篇文章 0 订阅

订阅专栏

数学

229 篇文章 0 订阅

订阅专栏

省选真题

115 篇文章 0 订阅

订阅专栏

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

令Ei=Fi/qi，求Ei.

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

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

构造数列

- 1 ( n - 1 ) 2, - 1 ( n - 2 ) 2, . . ., - 1 1 2, 0, 1 1 2, . . ., 1 ( n - 1 ) 2

$-\frac 1{(n-1)^2},-\frac 1{(n-2)^2},...,-\frac 1{1^2},0,\frac1{1^2},...,\frac 1{(n-1)^2}$
发现它和

qi ${q_i}$ 数列卷积之后就可以得到答案。

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
const double pi=acos(-1);
struct com
{
    double a,b;
    com operator + (const com &c) const
    {
        return (com){a+c.a,b+c.b};
    }
    com operator - (const com &c) const
    {
        return (com){a-c.a,b-c.b};
    }
    com operator * (const com &c) const
    {
        return (com){a*c.a-b*c.b,a*c.b+b*c.a};
    }
}a[400010],b[400010],p[400010];
int n,t,l,to[400010];
int rev(int x)
{
    int i,ret=0;
    for (i=0;i<t;i++)
        ret|=((x>>i)&1)<<t-i-1;
    return ret;
}
void init()
{
    int i;
    scanf("%d",&n);
    for (i=0;i<n;i++)
        scanf("%lf",&a[i].a);
    for (i=1;i<n;i++)
        b[n-i-1].a=-1.0/i/i,b[n+i-1].a=1.0/i/i;
    t=0,l=1;
    while (l<2*n-1) t++,l<<=1;
    p[0].a=1;
    p[1]=(com){cos(2*pi/l),sin(2*pi/l)};
    for (i=2;i<l;i++)
    {
        p[i]=p[i>>1]*p[i>>1];
        if (i&1) p[i]=p[i]*p[1];
    }
    for (i=0;i<l;i++)
        to[i]=rev(i);
}
void fft(com *a,int c)
{
    int i,j,k,w;
    com u,v;
    for (i=0;i<l;i++)
        if (to[i]>i)
            swap(a[i],a[to[i]]);
    for (i=1;i<l;i<<=1)
        for (j=0;j<l;j+=(i<<1))
        {
            w=0;
            for (k=j;k<j+i;k++)
            {
                u=a[k];
                v=a[k+i]*p[w];
                a[k]=u+v;
                a[k+i]=u-v;
                w+=c*(l/i>>1);
                if (w<0) w+=l;
            }
        }
}
void solve()
{
    int i;
    fft(a,1);
    fft(b,1);
    for (i=0;i<l;i++)
        a[i]=a[i]*b[i];
    fft(a,-1);
    for (i=n-1;i<2*n-1;i++)
    //for (i=0;i<l;i++)
        printf("%f\n",a[i].a/l);
}
int main()
{
    init();
    solve();
}