码迷,mamicode.com
首页 > 其他好文 > 详细

UR#34. 多项式乘法

时间:2017-01-08 16:38:14      阅读:239      评论:0      收藏:0      [点我收藏+]

标签:swap   print   head   输入   book   info   .net   element   ogg   

#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。

限制与约定

0n,m1050≤n,m≤105,保证输入中的系数大于等于 00 且小于等于 99。

时间限制1s1s

空间限制256MB256MB

下载

样例数据下载

#include<cmath>
#include<cstdio>
#include<complex>
using namespace std;
typedef complex<double> E;
const double Pi=acos(-1);
const int N=3e5+10;
int n,m,L,R[N];
E a[N],b[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 p=i<<1,j=0;j<n;j+=p){
            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;
            }
        }
    }
}
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;
    for(m+=n,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++) printf("%d ",(int)(a[i].real()/n+0.5));
    return 0;
}
/*
比着敲完就算了,FFT理解,来日方长。
zky神犇写的非常详细
http://blog.csdn.net/iamzky/article/details/22712347 
*/

 

UR#34. 多项式乘法

标签:swap   print   head   输入   book   info   .net   element   ogg   

原文地址:http://www.cnblogs.com/shenben/p/6262139.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!