【ZJOI2014】bzoj3527 力

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(n1)2,1(n2)2,...,112,0,112,...,1(n1)2

发现它和 qi 数列卷积之后就可以得到答案。

#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();
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值