[bzoj2179]FFT快速傅立叶

2179: FFT快速傅立叶

Time Limit: 10 Sec  Memory Limit: 259 MB
[Submit][Status][Discuss]

Description

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

Input

第一行一个正整数n。 第二行描述一个位数为n的正整数x。 第三行描述一个位数为n的正整数y。

Output

输出一行，即x*y的结果。

Sample Input

1
3
4

Sample Output

12

数据范围：
n<=60000

HINT

fft，裸的

这个板子真的好，相对于丁神的，优化了一些代码长度

抽时间讲一下fft

#include<algorithm>#include<iostream>#include<complex>#include<cstdio>#include<cmath>#define pi acos(-1)using namespace std;const int N = 200000 + 5;typedef complex<double> E;E a[N],b[N]; char ch[N];int n,m,L,c[N],r[N];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 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 ) for( int i = 0; i < n; i++ ) a[i] /= n;}int main(){scanf("%d", &n); n--;scanf("%s", ch); for( int i = 0; i <= n; i++ ) a[i] = ch[n-i]-'0';scanf("%s", ch); for( int i = 0; i <= n; i++ ) b[i] = ch[n-i]-'0';m = 2*n; for( n = 1; n <= m; n <<= 1 ) L++;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 = 0; i <= m; i++ )if( c[i] >= 10 ){c[i+1] += c[i]/10, c[i] %= 10;if( i == m ) m++;}for( int i = m; i >= 0; i-- ) printf("%d", c[i]);return 0;}