bzoj2179 FFT快速傅立叶

Description

---

给出两个n位10进制整数x和y，你需要计算x*y。

n<=60000

Solution

---

放一波板子，计算卷积利器，一个极佳blog
大概就是把系数表示法变成点值表示法，这样就直接扫一遍乘起来即可。可以理解成用一个函数去拟合之类的，对于一个n次的多项式我们需要取n-1个点
这里实际上利用了复数的各种神奇性质使得我们可以取值插值用分治做到nlogn，但是由于需要补全高位上的0项使得常数巨大，而且我打的还是递归-_-||

Code

---

#include <stdio.h>
#include <math.h>
#include <string.h>
#include <algorithm>
#include <complex>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define drp(i,st,ed) for (int i=st;i>=ed;--i)
using namespace std;
typedef complex<double> com;
const double Pi=3.14159265358979323846264338327950288;
const int N=131082;
char sta[N],stb[N];
com a[N],b[N];
int prt[N];
int read() {
    int x=0,v=1; char ch=getchar();
    for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
    for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
    return x*v;
}
void FFT(com *c,int len,int f) {
    if (len==1) return ;
    com wn(cos(2.0*Pi/len),sin(f*2.0*Pi/len)),w(1,0),t;
    com c1[len/2],c2[len/2];
    rep(i,0,len/2-1) {
        c1[i]=c[i*2+0];
        c2[i]=c[i*2+1];
    }
    FFT(c1,len/2,f); FFT(c2,len/2,f);
    for (int i=0;i<len/2;i++,w*=wn) {
        t=c2[i]*w;
        c[i]=c1[i]+t;
        c[i+len/2]=c1[i]-t;
    }
}
int main(void) {
    scanf("%*d%s%s",sta,stb);
    int n=strlen(sta)-1,m=strlen(stb)-1;
    rep(i,0,n) a[i]=sta[i]-'0';
    rep(i,0,m) b[i]=stb[i]-'0';
    m+=n; for (n=1;n<=m;n<<=1);
    FFT(a,n,1); FFT(b,n,1);
    rep(i,0,n) a[i]*=b[i];
    FFT(a,n,-1);
    int tmp=0;
    drp(i,m,0) {
        prt[i]=(tmp+(int)(a[i].real()/n+0.5))%10;
        tmp=(tmp+(int)(a[i].real()/n+0.5))/10;
    }
    if (tmp) printf("%d",tmp);
    rep(i,0,m) printf("%d",prt[i]);
    return 0;
}