bzoj2194 FFT

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

lshacm

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

阅读量338

点赞数

分类专栏：数学

本文链接：https://blog.csdn.net/u013514182/article/details/45503501

版权

数学专栏收录该内容

9 篇文章 0 订阅

订阅专栏

题意：计算C[k]=sigma(a[i]*b[i-k]) 其中 k < = i < n ，并且有 n < = 10 ^ 5。 a,b中的元素均为小于等于100的非负整数。
解法：FFT
1.将a数组反转，C[k]=sigma(a[n-1-i]*b[i-k]) 其中 k < = i < n。(n-1-i)+(i-k)=n-1-k。若把a[i]看做多项式A幂次为i的系数，则C[k]为多项式C幂次为n-1-k前的系数
2.求反向数组a和数组b构成多项式的乘积
3.倒序输出乘积多项式幂次为n-1,n-2…0前的系数

注意：代码中n变量在进行FFT操作时已修改，因此必须使用m/2表示之前的变量n！

#include <stdio.h>
#include <string.h>
#include <set>
#include <map>
#include <algorithm>
#include<complex>
#include<iostream>
#define pi acos(-1)
using namespace std ;
typedef complex<double> E;

const int maxn = 100000+10;

E a[maxn],b[maxn];
int n,m,L;
int R[maxn];
int x,y;
int c[maxn];

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 p=i<<1,j=0;j<n;j+=p){
            E w(1,0);
            for(int k=0;k<i;k++,w*=wn){
                E x=a[j+k],y=w*a[j+k+i];
                a[j+k]=x+y; a[j+k+i]=x-y;
            }
        }
    }
    if(f==-1) for(int i=0;i<n;i++) a[i]/=n;
}

int main(){
    //freopen("a.txt","r",stdin);
    scanf("%d",&n);
    for(int i=0;i<n;i++) { scanf("%d%d",&x,&y);
        E tmpx(x,0),tmpy(y,0);
    a[n-1-i]=tmpx,b[i]=tmpy; }
    L=0,m=2*n;
    for(n=1;n<=m;n<<=1) L++;//n->第一个大于m的2的指数 L->对应2的幂次
    for(int i=0;i<n;i++) R[i]=(R[i>>1]>>1)|((i&1)<<(L-1));
    fft(a,1),fft(b,1);
    for(int i=0;i<=n;i++) a[i]*=b[i];
    fft(a,-1);
    for(int i=0;i<=m;i++) c[i]=(int)(a[i].real()+0.1);
    for(int i=m/2-1;i>=0;i--) printf("%d\n",c[i]);//n的值发生变化，不能再使用n作为原来的变量使用！
    return 0;
}