BZOJ2194: 快速傅立叶之二

最新推荐文章于 2020-09-21 22:30:57 发布

CR1SceNT

最新推荐文章于 2020-09-21 22:30:57 发布

阅读量525

点赞数 1

分类专栏： FFT

本文链接：https://blog.csdn.net/ep1c_heret1c/article/details/70823263

版权

FFT 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

BZOJ2194

中午看着题解打完了。。晚上才想清白。。
题目都说了fft肯定就把式子往卷积上转啦。
然而题目中是求下标差一定的两个多项式的乘积和，那么就把 $a$ 或者 $b$ 任意反一个就好了。
假设反 $b$
那么 $C[k]=\sum_{i=k}^{n-1}{a[i]*b[n+k-i]}$
令 $D[k]=\sum_{i=k}^{n-1}{a[i]*b[k-i]}$
那么 $C[k]=D[n+k]$
发现求 $Dk$ 的式子很像卷积啊，然而正常卷积不应该是 $\sum_{i=0}^k$ 吗。。
就是这个东西纠结了我很久。。其实把对应的项写出来。。发现其余的都是0毫无影响。。
$D[n+k]=\sum_{i=0}^{n+k}{a[i]*b[n+k-i]}$

【代码】

#include <cstdio>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <complex>
#include <algorithm>
#include <cmath>
#define N 262145
#define INF 0x7fffffff
using namespace std;
typedef complex<double> C;
const double pi=acos(-1);

int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

int n,m;
C a[N],b[N];

void FFT(C *a,int n,int f)
{
    if(n==1) return;
    C wn(cos(2*pi/n),sin(2*pi*f/n)),w(1,0),t;
    C a0[n>>1],a1[n>>1];
    for(int i=0;i<n>>1;i++) a0[i]=a[i<<1],a1[i]=a[i<<1|1];
    FFT(a0,n>>1,f),FFT(a1,n>>1,f);
    for(int i=0;i<n>>1;i++,w*=wn)
    {
        t=a1[i]*w;
        a[i]=a0[i]+t;
        a[i+(n>>1)]=a0[i]-t;
    }
}

int main() 
{
    n=m=read()-1;
    for(int i=0;i<=n;i++) 
    {
        a[i]=read();
        b[n-i]=read();
    }
    m+=n;
    for(n=1;n<=m;n<<=1);
    FFT(a,n,1);FFT(b,n,1);
    for(int i=0;i<=n;i++) a[i]*=b[i];
    FFT(a,n,-1);
    for(int i=m>>1;i<=m;i++) printf("%d\n",int(a[i].real()/n+0.5));
    return 0;
}