這是一道模板題。

給你兩個多項式，請輸出乘起來後的多項式。

輸入格式

第一行兩個整數 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>
#include<algorithm>
#include<cmath>
#define swap(x,y) x ^= y, y ^= x, x ^= y
#define LL long long 
using namespace
 
 std;
const int MAXN = 3 * 1e6 + 10;
inline int read(){
    int x=0,f=1;char ch=‘ ‘;
    while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();}
    while(ch>=‘0‘&&ch<=‘9‘)x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
    return x*f;
}
int N, M, limit = 1, L;
const int P = 998244353
 
, G = 3, Gi = 332748118; 
LL a[MAXN], b[MAXN];
int r[MAXN];
inline LL fastpow(LL a, LL k) {
    LL base = 1;
    while(k) {
        if(k & 1) base = (base * a ) % P;
        a = (a * a) % P;
        k >>= 1;
    }
    return base % P;
}
inline void NTT(LL *A, int type) {
    for(int i = 0; i < limit; i++) 
        if(i < r[i]) swap(A[i], A[r[i]]);
    for(int mid = 1; mid < limit; mid <<= 1) {    
        LL Wn = fastpow( type == 1 ? G : Gi , (P - 1) / (mid << 1));
        for(int j = 0; j < limit; j += (mid << 1)) {
            LL w = 1;
            for(int k = 0; 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() {
    #ifdef WIN32
    freopen("a.in", "r", stdin);
    #endif
    N = read(); M = read();
    for(int i = 0; i <= N; i++) a[i] = (read() + P) % P;
    for(int i = 0; i <= M; i++) b[i] = (read() + P) % P;
    while(limit <= N + M) limit <<= 1, L++;
    for(int i = 0; i < limit; i++)
        r[i] = (r[i >> 1] >> 1) | ((i & 1) << (L - 1));    
    NTT(a, 1);NTT(b, 1);    
    for(int i = 0; i < limit; i++) a[i] = (a[i] * b[i]) % P;
    NTT(a, -1);    
    LL inv = fastpow(limit, P - 2);
    for(int i = 0; i <= N + M; i++)
        printf("%d ", (a[i] * inv) % P);
    return 0;
}

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