【BZOJ】2179: FFT快速傅立叶（fft）

#include <cstdio>
#include <cstring>
#include <cmath>
#include <string>
#include <iostream>
#include <algorithm>
#include <queue>
#include <set>
#include <map>
using namespace std;
typedef long long ll;
#define rep(i, n) for(int i=0; i<(n); ++i)
#define for1(i,a,n) for(int i=(a);i<=(n);++i)
#define for2(i,a,n) for(int i=(a);i<(n);++i)
#define for3(i,a,n) for(int i=(a);i>=(n);--i)
#define for4(i,a,n) for(int i=(a);i>(n);--i)
#define CC(i,a) memset(i,a,sizeof(i))
#define read(a) a=getint()
#define print(a) printf("%d", a)
#define dbg(x) cout << (#x) << " = " << (x) << endl
#define error(x) (!(x)?puts("error"):0)
inline const int getint() { int r=0, k=1; char c=getchar(); for(; c<‘0‘||c>‘9‘; c=getchar()) if(c==‘-‘) k=-1; for(; c>=‘0‘&&c<=‘9‘; c=getchar()) r=r*10+c-‘0‘; return k*r; }
#define rdm(x, i) for(int i=ihead[x]; i; i=e[i].next)

struct cp {
	double r, i;
	cp(double _r=0.0, double _i=0.0) : r(_r), i(_i) { }
	cp operator+ (const cp &x) const { return cp(r + x.r, i + x.i); }
	cp operator- (const cp &x) const { return cp(r - x.r, i - x.i); }
	cp operator* (const cp &x) const { return cp(r*x.r - i*x.i, r*x.i + i*x.r); }
};
const int N=200005;
const double PI=acos(-1.0);
cp A[N];
int rev[N];
void dft(cp *a, int n, int flag) {
	rep(i, n) A[rev[i]]=a[i];
	rep(i, n) a[i]=A[i];
	for(int m=2; m<=n; m<<=1) {
		cp wn(cos(2.0*PI/m*flag), sin(2.0*PI/m*flag));
		for(int i=0; i<n; i+=m) {
			cp w(1.0); int mid=m>>1;
			for(int j=0; j<mid; ++j) {
				cp u=a[i+j+mid]*w, t=a[i+j];
				a[i+j]=t+u;
				a[i+j+mid]=t-u;
				w=w*wn;
			}
		}
	}
	if(flag==-1) rep(i, n) a[i].r/=n;
}
void init(int &len) {
	int k=1, L=0;
	while(k<len) k<<=1, ++L;
	len=k;
	rep(i, len) {
		int t=i, r=0; k=L;
		while(k--) r<<=1, r|=(t&1), t>>=1;
		rev[i]=r;
	}
}
int len, ans[N];
char s[N];
cp a[N], b[N];
void readin(cp *a) {
	scanf("%s", s);
	int n=strlen(s);
	rep(i, n) a[i].r=s[n-i-1]-‘0‘;
}
int main() {
	read(len); len=len*2-1;
	readin(a); readin(b);
	init(len);
	dft(a, len, 1); dft(b, len, 1);
	rep(i, len) a[i]=a[i]*b[i];
	dft(a, len, -1);
	rep(i, len) ans[i]=a[i].r+0.5;
	rep(i, len) ans[i+1]+=ans[i]/10, ans[i]%=10;
	++len;
	while(!ans[len] && len) --len;
	for3(i, len, 0) printf("%d", ans[i]);
	return 0;
}