UOJ34. 多项式乘法
FFT解法
前方高能,你需要(看一些[算导]涨姿势)
答记者问
Q:FFT是什么东东?
A:自己看算导!!!
Q:(看算导,提高了姿势水平后)蝴蝶变换怎么办办啊
A:实在不行你就背吧
Q:自己打的版?
A:hzwer大佬的,这是链接,两个模板哦。
Q:这个解释我们新闻界不懂啊。。。
A:(其实我也是一知半解不敢说全懂=-=)zky神犇写的非常详细。。。
http://blog.youkuaiyun.com/iamzky/article/details/22712347
先自己试试?
uoj 34
Sample Input
1 2
1 2
1 2 1
Sample Output
1 4 5 2
我这么说你们可能不懂,看代码!!!
总结:
1.FFT (把系数表示变成点集表示,就是函数->函数上的点)
2.卷积(A函数上的点*B函数上的点=C函数上的点)
3.反FFT(插值,C的点集表示变成系数表示)
下发文件
//uoj34
#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<math.h>
#include<cctype>
#include<complex>
#define pi acos(-1)
#define N 300000
#define LL long long
#define E complex <double>
using namespace std;
int n, m, L, t, w;
int R[N], c[N];
E a[N], b[N];
int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
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;
}
void init()
{
n = read(); t = read();
w = n+t;
for(int i = 0,j; i <= n; i++) j = read(),a[n-i] = j;
for(int i = 0,j; i <= t; i++) j = read(),b[t-i] = j;
return ;
}
void work()
{
n = max(n, t);
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);//FFT
for(int i=0; i<=n; i++) a[i]*=b[i];//卷积?-?
fft(a, -1);//反FFT
for(int i=0; i<=n; i++) c[i] = (int) (a[i].real() + 0.1);
//精度问题,解决因为精度出现0.99的尴尬问题
/*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=w; i>=0; i--) printf("%d ",c[i]); //本题不去前导0
return ;
}
int main()
{
init();
work();
return 0;
}
人生经验
如果你解决本题有什么人生经验(程序错误,我的错误,不懂的地方,奇奇怪怪的优化想法)可以留言偶!,我会选一些挂上。