UOJ#34(FFT模板)

题目链接:http://uoj.ac/problem/34

思路:FFT模板。

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll inff = 0x3f3f3f3f3f3f3f3f;
#define FOR(i,a,b) for(int i(a);i<=(b);++i)
#define FOL(i,a,b) for(int i(a);i>=(b);--i)
#define REW(a,b) memset(a,b,sizeof(a))
#define inf int(0x3f3f3f3f)
#define si(a) scanf("%d",&a)
#define sl(a) scanf("%I64d",&a)
#define sd(a) scanf("%lf",&a)
#define ss(a) scanf("%s",a)
#define mod ll(998244353)
#define pb push_back
#define lc d<<1
#define rc d<<1|1
#define Pll pair
#define P pair
#define pi acos(-1)
int n,m;
struct CP{
    double x,y;
    CP(){} CP(double a,double b):x(a),y(b){}
    CP operator+(const CP&r) const{return CP(x+r.x,y+r.y);}
    CP operator-(const CP&r) const{return CP(x-r.x,y-r.y);}
    CP operator*(const CP&r) const{return CP(x*r.x-y*r.y,x*r.y+y*r.x);}
}a[300008],b[300008];
inline void fft(CP a[],int f) 
{
    int i,j,k;
    for(i=j=0;ij) swap(a[i],a[j]);
        for(k=n>>1;(j^=k)>=1);
    }
    for (int d=0;(1<>n>>m;
    FOR(i,0,n) sd(a[i].x);
    FOR(i,0,m) sd(b[i].x);
    for(m=n+m,n=1;n<=m;n<<= 1);
    fft(a,1),fft(b,1);
    FOR(i,0,n) a[i]=a[i]*b[i];
    fft(a,-1);
    FOR(i,0,m) printf("%d ",(int)(a[i].x+0.1));
    return 0;
}

 

你可能感兴趣的:(数论)