[挖坑][uoj]多项式乘法 FFT

#34. 多项式乘法

 统计

 描述
 提交
 自定义测试

这是一道模板题。
给你两个多项式，请输出乘起来后的多项式。

输入格式

第一行两个整数 nn 和 mm，分别表示两个多项式的次数。
第二行 n+1n+1 个整数，分别表示第一个多项式的 00 到 nn 次项前的系数。
第三行 m+1m+1 个整数，分别表示第一个多项式的 00 到 mm 次项前的系数。

输出格式

一行 n+m+1n+m+1 个整数，分别表示乘起来后的多项式的 00 到 n+mn+m 次项前的系数。

样例一

input

1 2
1 2
1 2 1

output

1 4 5 2

explanation

(1+2x)⋅(1+2x+x2)=1+4x+5x2+2x3(1+2x)⋅(1+2x+x2)=1+4x+5x2+2x3。

限制与约定

0≤n,m≤1050≤n,m≤105，保证输入中的系数大于等于 00 且小于等于 99。
时间限制：1s1s
空间限制：256MB
由于博主是个智障，数学没学好， 复数都不会，所以此题仅仅是挖坑，抄了下模板
等到noip，我一定会更新（AFO）的
#include<bits/stdc++.h>
#define pi acos(-1)
using namespace std;
typedef complex<double> E;
int n,m;
E a[262145],b[262145];
void fft( E *x, int n, int type ){
if( n == 1 ) return ;
E l[n>>1], r[n>>1];
for( int i = 0; i < n; i += 2 ) l[i>>1] = x[i], r[i>>1] = x[i+1];
fft( l, n>>1, type ); fft( r, n>>1, type );
E wn(cos(2*pi/n),sin(type*2*pi/n)),w(1,0),t;
for( int i = 0; i < n>>1; i++, w *= wn )
t = w*r[i], x[i] = l[i]+t, x[i+(n>>1)] = l[i]-t;
}
int main(){
scanf("%d%d", &n, &m);
for( int i = 0,x; i <= n; i++ ) scanf("%d", &x), a[i] = x;
for( int i = 0,x; i <= m; i++ ) scanf("%d", &x), b[i] = x;
m = n+m; for( n = 1; n <= m; n <<= 1 );
fft( a, n, 1 ); fft( b, n, 1 );
for( int i = 0; i <= n; i++ ) a[i] *= b[i];
fft( a, n, -1 );
for( int i = 0; i <= m; i++ ) printf("%d ", (int)(a[i].real()/n+0.5));
return 0;
}