【模板】多项式乘法（FFT/NTT）

题解:https://www.luogu.org/problemnew/show/P3803

题解:直接做是n^2的。可以用FFT优化。首先要知道一些知识。

一.单位复数根

定义:。

表示单位复数根，相当于将复平面的单位圆等分成n份(n一般是2的多少次幂)，从实数轴逆时针旋转对应的坐标。

性质：,。还是很容易推出来的。

二.点值表示法

一个n次多项式可以用n个坐标来代替。我们用单位复数根来当做坐标，要求出函数值。设:

奇偶分类:

将w(k,n)代入看、k<n/2

将w(k+n/2,n)代入

注意到上面两个式子只有一个符号不同，所以通过枚举第一个式子可以O(1)的求出另一个，这样规模就减半了。

三.逆变换

设:。。

四.迭代实现

通过观察性质，反转后的序列就是原序列的下标的二进制反转。我们用数位dp的思想。r[i]表示将i翻转后的下标。例如11001->10011。其实就是将最右边的1移到最高位，剩下的1100已经在之前做过了，直接将两部分合并就行了。

 

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
const double pi=acos(-1);
struct cp{
	double r,i;
	cp(){}
	cp(double x,double y){
		r=x;i=y;
	}
	cp operator +(const cp &a) const {
		return cp(r+a.r,i+a.i);
	} 
	cp operator - (const cp &a) const{
		return cp(r-a.r,i-a.i);
	}
	cp operator * (const cp &a)const {
		return cp(r*a.r-i*a.i,r*a.i+i*a.r);
	}
}a[N<<2],b[N<<2];
int R[N<<2];//反码 
int n,m,L;

inline void fft(cp *a,int len,int op){
	
	for(int i=0;i<len;i++) {
		if(i<R[i]) swap(a[ i ],a[ R[i] ]);
	}
	for(int i=2;i<=len;i<<=1){// 
		cp wn=cp(cos(2*pi/i), op*sin(2*pi/i));//单位复数根 
		for(int j=0;j<len;j+=i){
			cp w=cp(1.0,0.0);
			for(int k=j;k<j+(i/2);k++,w=w*wn){
				cp u=a[k];
				cp v=w*a[k+i/2]; 
				a[k]=u+v;
				a[k+i/2]=u-v;
			}
			
		}
		
		
	}
	if(op==-1) for(int i=0;i<len;i++) a[i].r/=len;
}


int main(){
//	cp a=cp(1.0,2.0);
//	cp b=cp(1.0,2.0);
//	cp c=a*b;
	scanf("%d%d",&n,&m);
	for(int i=0;i<=n;i++) scanf("%lf",&a[i].r);
	for(int i=0;i<=m;i++) scanf("%lf",&b[i].r);
	
	// 初始化 
	m+=n;n=1;L=0;
	for(;n<=m;n<<=1)L++;
    for(int i=0;i<n;i++) R[i]=(R[i>>1]>>1)|(i&1)<<(L-1);
  
    fft(a,n,1);fft(b,n,1);
    for(int i=0;i<n;i++) a[i]=a[i]*b[i];
    fft(a,n,-1);

	for(int i=0;i<=m;i++) printf("%d ",(int)(a[i].r+0.5));//四舍五入 
		
	return 0;
} 

/*
3 4
1 2 3 4
1 2 3 4 5 

//0 4 2 6 1 5 3 7 R[i] n=8 

*/