LibreOJ #108. 多项式乘法

二次联通门 : LibreOJ #108. 多项式乘法

/*
    LibreOJ #108. 多项式乘法

    FFT板子题
    不行啊。。。跑的还是慢

    应该找个机会学一学由乃dalao的fft
    或者是毛爷爷的fft，跑的真是快啊。。。
*/
#include <cstdio>
#include <iostream>
#include <cmath>

const int BUF = ;
char Buf[BUF], *buf = Buf;

inline void read (int &now)
{
    for (now = ; !isdigit (*buf); ++ buf);
    for (; isdigit (*buf); now = now *  + *buf - '', ++ buf);
}
using std :: swap;
#define Max 3000000
typedef double flo;
struct Vec
{
    flo r, i; Vec () {}
    Vec (flo x, flo y) : r (x), i (y) {}
    Vec operator * (const Vec &b) const
    { return Vec (r * b.r - i * b.i, r * b.i + i * b.r); }
    Vec operator * (const flo &k) const
    { return Vec (r * k, i * k); }
    Vec operator + (const Vec &b) const
    { return Vec (r + b.r, i + b.i); }
    Vec operator - (const Vec &b) const
    { return Vec (r - b.r, i - b.i); }
    Vec& operator /= (const flo &k)
    { return r /= k, i /= k, *this; }
};

Vec a[Max], b[Max];
int N, M, Maxn, rader[Max];
const flo PI = acos (-);

void FFT (Vec *a, int N, int f = )
{
    register int i, j, k;
    for (i = ; i < N; ++ i)
        if (rader[i] > i) swap (a[i], a[rader[i]]);
    for (k = ; k < N; k <<= )
    {
        Vec wn (cos (PI / k), f * sin (PI / k));
        for (j = ; j < N; j += k << )
        {
            Vec w (, ), t;
            for (i = j; i < j + k; ++ i, w = w * wn)
            {
                t = w * a[i + k];
                a[i + k] = a[i] - t;
                a[i] = a[i] + t;
            }
        }
    }
    if (f == -)
        for (i = ; i < N; ++ i) a[i] /= N;
}

int Main ()
{
    fread (buf, , BUF, stdin);
    read (N), read (M); register int i; int x;
    ++ N, ++ M, Maxn =  << int (ceil (log2 (N + M)));
    for (i = ; i < N; ++ i) read (x), a[i].r = x;
    for (i = ; i < M; ++ i) read (x), b[i].r = x;

    for (i = ; i < Maxn; ++ i)
       rader[i] = rader[i >> ] >>  | (i & ) * (Maxn >> );
    FFT (a, Maxn), FFT (b, Maxn);
    for (i = ; i < Maxn; ++ i)
        a[i] = a[i] * b[i];
    N = N + M - ;
    for (FFT (a, Maxn, -), i = ; i <= N; ++ i)
        printf ("%d ", int (round (a[i].r)));
    return ;
}
int ZlycerQan = Main ();
int main (int argc, char *argv[]) {;}