BZOJ 2194 快速傅立叶之二

本文介绍了一种使用快速傅立叶变换(FFT)解决特定数学问题的方法。通过将序列进行翻转和构造,使得原本难以直接应用FFT的问题得以解决。文章详细展示了如何利用FFT计算两个序列的卷积,并给出了具体的实现代码。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目大意

已知 a,ba,b 序列,计算 ck=ai×bikck=∑ai×bi−k

题解

观察题目名称,可以想到FFT……
FFT能解决的是形如下面的式子:
hk=fi×gkihk=∑fi×gk−i
可以发现,ff 数组的下标和 g 数组的下标和不变,且恒为 kk
小小地转换一下,把 b 数组前后翻转,得:
ck=ai×bn1(ik)=ai×bn1i+kck=∑ai×bn−1−(i−k)=∑ai×bn−1−i+k
这样,和不变,恒为 n1+kn−1+k。然后再构造 ff 序列:
ck=ai×bn1i+k=fn1+k
用FFT计算 ff 序列,然后再输出相应的 c 序列就好了。

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=2148576;
const double pi=acos(-1.0);
struct comp{
    double x,y;
    comp(double xx=0,double yy=0):x(xx),y(yy) {}
    friend comp operator+(const comp &x,const comp &y) {return comp(x.x+y.x,x.y+y.y);}
    friend comp operator-(const comp &x,const comp &y) {return comp(x.x-y.x,x.y-y.y);}
    friend comp operator*(const comp &a,const comp &b) {return comp(a.x*b.x-a.y*b.y,a.x*b.y+b.x*a.y);}
}a[maxn],b[maxn];
int limit=1,r[maxn];
void fft(comp *t,int ty){
    for(int i=0;i<limit;i++)
        if(i<r[i])
            swap(t[i],t[r[i]]);
    for(int mid=1;mid<limit;mid<<=1){
        comp wn(cos(pi/mid),ty*sin(pi/mid));
        for(int j=0,R=(mid<<1);j<limit;j+=R){
            comp w(1,0);
            for(int k=0;k<mid;k++,w=w*wn){
                comp x=t[j+k],y=w*t[j+k+mid];
                t[j+k]=x+y;
                t[j+k+mid]=x-y;
            }
        }
    }
}
int main(void)
{
    int n,l=0;
    scanf("%d",&n);
    while(limit<=(n<<1))
        limit<<=1,++l;
    for(int i=1;i<limit;i++)
        r[i]=((r[i>>1]>>1)|((i&1)<<(l-1)));
    for(int i=0;i<n;++i)
        scanf("%lf%lf",&a[i].x,&b[n-i-1].x);
    fft(a,1);
    fft(b,1);
    for(int i=0;i<limit;i++)
        a[i]=a[i]*b[i];
    fft(a,-1);
    for(int i=n-1;i<(n<<1)-1;i++)
        printf("%d\n",(int)(a[i].x/limit+0.5));
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值