UOJ#34. 多項式乘法(NTT)
阿新 • • 發佈:2018-05-02
put line fde var str 一個 時間 XP col
這是一道模板題。
給你兩個多項式,請輸出乘起來後的多項式。
輸入格式
第一行兩個整數 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 namespacestd; 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)