UOJ#34. 多项式乘法(NTT)

这是一道模板题。

给你两个多项式，请输出乘起来后的多项式。

输入格式

第一行两个整数 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。

限制与约定

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

时间限制：1s1s

空间限制：256MB

震惊!

TLE一上午的原因竟然是素数和原根的定义没有加const！

NTT的板子题

把单位元换成原根就好

#include<cstdio>
#define getchar() (p1 == p2 && (p2 = (p1 = buf) + fread(buf, 1, 1<<21, stdin), p1 == p2) ? EOF : *p1++)
#define swap(x,y) x ^= y, y ^= x, x ^= y
#define LL long long
const int MAXN =  * 1e6 + , P = , G = , Gi = ;
char buf[<<], *p1 = buf, *p2 = buf;
inline int read() {
    char c = getchar(); int x = , f = ;
    while(c < '' || c > '') {if(c == '-') f = -; c = getchar();}
    while(c >= '' && c <= '') x = x *  + c - '', c = getchar();
    return x * f;
}
int N, M, limit = , L, r[MAXN];
LL a[MAXN], b[MAXN];
inline LL fastpow(LL a, LL k) {
    LL base = ;
    while(k) {
        if(k & ) base = (base * a ) % P;
        a = (a * a) % P;
        k >>= ;
    }
    return base % P;
}
inline void NTT(LL *A, int type) {
    for(int i = ; i < limit; i++)
        if(i < r[i]) swap(A[i], A[r[i]]);
    for(int mid = ; mid < limit; mid <<= ) {
        LL Wn = fastpow( type ==  ? G : Gi , (P - ) / (mid << ));
        for(int j = ; j < limit; j += (mid << )) {
            LL w = ;
            for(int k = ; k < mid; k++, w = (w * Wn) % P) {
                 int x = A[j + k], y = w * A[j + k + mid] % P;
                 A[j + k] = (x + y) % P,
                 A[j + k + mid] = (x - y + P) % P;
            }
        }
    }
}
int main() {
    N = read(); M = read();
    for(int i = ; i <= N; i++) a[i] = (read() + P) % P;
    for(int i = ; i <= M; i++) b[i] = (read() + P) % P;
    while(limit <= N + M) limit <<= , L++;
    for(int i = ; i < limit; i++) r[i] = (r[i >> ] >> ) | ((i & ) << (L - ));
    NTT(a, );NTT(b, );
    for(int i = ; i < limit; i++) a[i] = (a[i] * b[i]) % P;
    NTT(a, -);
    LL inv = fastpow(limit, P - );
    for(int i = ; i <= N + M; i++)
        printf("%d ", (a[i] * inv) % P);
    return ;
}