标签:FFT
题目
Description 给出两个n位10进制整数x和y,你需要计算xy。 Input 第一行一个正整数n。 第二行描述一个位数为n的正整数x。 第三行描述一个位数为n的正整数y。 Output 输出一行,即xy的结果。 Sample Input 1
3
4
Sample Output 12
数据范围:
n<=60000
分析
FFT模板题,理解起来好困难啊,照着黄学长的代码敲的,看来我不适合学习FFT。。。
bzoj100题纪念!!!
快速傅里叶变换FFT*
复数和复平面
我们把形如a+bi(a,b均为实数)的数称为复数,其中a称为实部,b称为虚部,i称为虚数单位。当虚部等于零时,这个复数可以视为实数;当z的虚部不等于零时,实部等于零时,常称z为纯虚数。
其中i是方程x2=−1的根,相当于√−1(注意只是相当于,易于理解!!!)
将复数的实部与虚部的平方和的正的平方根的值称为该复数的模,记作∣z∣ 即对于复数z=a+bi,它的模为√a2+b2
称复数z′=a−bi为z的共轭复数。即两个实部相等,虚部互为相反数的复数互为共轭复数。
运算法则和普通多项式类似
复平面就是将z=a+bi在平面上表示为(a,b),x轴为实部单位1,y轴为虚部单位i
多项式的表达方式
系数表达和点值表达
系数表达就是大家常用的表达方式,点值表达就像在这个多项式函数上取n个不同的点,这样就可以确定原多项式。
举个栗子
我们通过三点可以确定一个二次函数,两点确定一条直线。一个n次多项式需要n个点(n次多项式意思是有0到n-1次幂的多项式)。
f(x)=x2+2x−1可以被表达为(0,−1),(1,2),(2,7)
将多项式用点值表达可以快速加法和乘法
单位根
这样算出来的点值表示法,那么对应的求值点究竟是哪些呢?
答案是2n次单位根。
数学上,n次单位根是n次幂为1的复数。它们位于复平面的单位圆上,构成正n边形的顶点,其中一个顶点是1。
复数中1恰好有n个单位根e2kπi/n
eix=cos x+i sin x
DFT和FFT
使用单位根计算点值表达式叫DFT(离散傅里叶变换)复杂度O(n2),FFT是其分治的优化版,复杂度O(n log n)
//以下部分看不懂的可以跳过(本部分引用自picks博客http://picks.logdown.com/posts/177631-fast-fourier-transform)
假如我们取单位根的幂进行转换,会有什么效果?
设A0(x)是A(x)的偶次项的和,A1(x)是奇次项的和。那么:
A(ωmn)=A0((ωmn)2)+ωmnA1((ωmn)2)=A0(ωmn2)+ωmnA1(ωmn2)A(ωm+n2n)=A0((ωmn)2)+ωm+n2nA1((ωmn)2)=A0(ωmn2)−ωmnA1(ωmn2)
即我们只要有了A0(x)A1(x)的点值表示,就能在O(n)时间内算出A(x)的点值表示。
code
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<complex>
#define pi acos(-1)
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#define mem(x,num) memset(x,num,sizeof x)
using namespace std;
inline ll read()
{
ll f=1,x=0;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
const int maxn=2e5+6;
typedef complex<double> E;
int n,m,L;
char ch[maxn];
int R[maxn],c[maxn];
E a[maxn],b[maxn];
void fft(E *a,int f)
{
rep(i,0,n-1)
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 j=0;j<n;j+=(i<<1)){
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)rep(i,0,n-1)a[i]/=n;
}
int main()
{
n=read();n--;
scanf("%s",ch);
rep(i,0,n)a[i]=ch[n-i]-'0';
scanf("%s",ch);
rep(i,0,n)b[i]=ch[n-i]-'0';
m=2*n;
for(n=1;n<=m;n<<=1)L++;
rep(i,0,n-1)R[i]=(R[i>>1]>>1)|((i&1)<<(L-1));
fft(a,1);fft(b,1);
rep(i,0,n)a[i]*=b[i];
fft(a,-1);
rep(i,0,m)c[i]=(int)(a[i].real()+0.1);
rep(i,0,m)
if(c[i]>=10){
c[i+1]+=c[i]/10,c[i]%=10;
if(i==m)m++;
}
dep(i,m,0)printf("%d",c[i]);
return 0;
}