【BZOJ 2194】 快速傅立叶之二|FFT
2016-03-23 15:37
525 查看
倒过来就是卷积了 就可以FFT了
弱智错误
1. /2
2. +0.5
3. <n
4. MAXN*4
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
using namespace std;
const double PI=acos(-1);
const int MAXN=400000+10;
struct C
{
double r,i;
C(double a=0,double b=0){r=a;i=b;}
C operator + (C x) {return C(x.r+r,x.i+i);}
C operator - (C x) {return C(r-x.r,i-x.i);}
C operator * (C x) {return C(r*x.r-i*x.i,x.i*r+x.r*i);}
};
int n,rev[MAXN];C a[MAXN],b[MAXN];
void FFT(C x[],int T)
{
for(int i=0;i<n;i++) if(rev[i]<i) swap(x[i],x[rev[i]]);
for(int i=2;i<=n;i<<=1)
{
C wn(cos(2*PI/i),T*sin(2*PI/i));
for(int j=0;j<n;j+=i)
{
C t,tmp,w(1,0);
for(int k=0;k<i/2;k++)
{
tmp=x[j+k];
t=w*x[j+k+i/2];
x[j+k]=tmp+t;
x[j+k+i/2]=tmp-t;
w=w*wn;
}
}
}
}
int main()
{
cin >>n; int tmp=n;
for(int i=0;i<n;i++) scanf("%lf %lf",&a[i].r,&b[n-i-1].r);
int m=n+n+1;n=1;
while(n<=m) n<<=1;
for(int i=0;i<n;i++) rev[i]= (rev[i>>1]>>1)|((i&1)?n>>1:0);
FFT(a,1);FFT(b,1);
for(int i=0;i<n;i++) a[i]=a[i]*b[i];
FFT(a,-1);
for(int i=0;i<tmp;i++) printf("%d\n",(int)(a[i+tmp-1].r/n+0.5));
return 0;
}
弱智错误
1. /2
2. +0.5
3. <n
4. MAXN*4
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
using namespace std;
const double PI=acos(-1);
const int MAXN=400000+10;
struct C
{
double r,i;
C(double a=0,double b=0){r=a;i=b;}
C operator + (C x) {return C(x.r+r,x.i+i);}
C operator - (C x) {return C(r-x.r,i-x.i);}
C operator * (C x) {return C(r*x.r-i*x.i,x.i*r+x.r*i);}
};
int n,rev[MAXN];C a[MAXN],b[MAXN];
void FFT(C x[],int T)
{
for(int i=0;i<n;i++) if(rev[i]<i) swap(x[i],x[rev[i]]);
for(int i=2;i<=n;i<<=1)
{
C wn(cos(2*PI/i),T*sin(2*PI/i));
for(int j=0;j<n;j+=i)
{
C t,tmp,w(1,0);
for(int k=0;k<i/2;k++)
{
tmp=x[j+k];
t=w*x[j+k+i/2];
x[j+k]=tmp+t;
x[j+k+i/2]=tmp-t;
w=w*wn;
}
}
}
}
int main()
{
cin >>n; int tmp=n;
for(int i=0;i<n;i++) scanf("%lf %lf",&a[i].r,&b[n-i-1].r);
int m=n+n+1;n=1;
while(n<=m) n<<=1;
for(int i=0;i<n;i++) rev[i]= (rev[i>>1]>>1)|((i&1)?n>>1:0);
FFT(a,1);FFT(b,1);
for(int i=0;i<n;i++) a[i]=a[i]*b[i];
FFT(a,-1);
for(int i=0;i<tmp;i++) printf("%d\n",(int)(a[i+tmp-1].r/n+0.5));
return 0;
}
相关文章推荐
- 花瓶 暴力
- 第一次使用less记录
- Android_Handler详解(一)
- 【bzoj2724】[Violet 6]蒲公英 (注意:题面有毒!)
- 并查集的初步学习
- 面试经验
- IOS开发 - 网络总结(一)
- 设计模式之外观模式
- EF 操作 视频教程
- 文字垂直滚动效果
- 精益生产管理
- Multiply Strings
- Android 最火的快速开发框架XUtils
- 带你快速了解 Beyond Compare
- Facebook广告API系列 Business Manager
- 练习1:第一个程序
- ListView的扩展
- Realsense(四)github上一种提取彩色和深度视频流并转为mat的方案
- Dorothy Lua开发建议
- 微信公众平台测试号