提交记录 16801


用户 题目 状态 得分 用时 内存 语言 代码长度
Saisyc 1002i. 【模板题】多项式乘法 Accepted 100 20.523 ms 7588 KB C++ 3.05 KB
提交时间 评测时间
2021-10-22 22:45:02 2021-10-22 22:45:05
#include <cstdio>
#include <cctype>
#include <algorithm>

namespace fastIO {
    const int SZ = 1 << 25;
    char ibuf[SZ], *p0 = ibuf, *p1 = ibuf;
    inline char getchar(void) { return p0 == p1 && (p1 = (p0 = ibuf) + fread(ibuf, 1, SZ, stdin), p0 == p1) ? EOF : *p0++; }
    char char_read; void read(auto &val) {
		val = 0; do char_read = getchar(); while(!isdigit(char_read));
		do val = val * 10 + char_read - '0'; while(isdigit(char_read = getchar()));
    }
    char obuf[SZ], *p = obuf;
    inline void putchar(char char_write) { *p++ = char_write; }
    void write(auto val) { if(val >= 10) write(val / 10); putchar(val % 10 + '0'); }
    inline void write(auto a[], int sz) { for(int i = 0; i < sz; ++i) write(a[i]), putchar(' '); putchar('\n'); }
    void output(void) { fwrite(obuf, p - obuf, 1, stdout);}
};

const int N =     1 << 21;
const int p =    81 << 21 | 1;
const int g =  1167 << 12 | 1;
const int h = 11443 << 12 | 1;

inline int pow_modp(int a, int b) { int x = 1; for(; b; a = (long long)a * a % p, b >>= 1) if(b & 1) x = (long long)x * a % p; return x; }

int ntt_sup[N];
inline void ntt(auto a[], int n, const int g) {
	auto ntt_a = ntt_sup, j = a, _j = a; int i, k, _k, w, _w;
	for(i = 1, w = pow_modp(g, (p - 1) / n); i < n; i <<= 1, w = (long long)w * w % p, std :: swap(a, ntt_a))
		for(k = 0, j = a, _j = a + (n >> 1), _w = 1; k != n; k += i, _w = (long long)_w * w % p)
			for(_k = k, k += i; _k != k; ++j, ++_j, ++_k)
				ntt_a[_k] = *j + *_j < 0 ? *j + *_j + p : *j + *_j - p,
				ntt_a[_k + i] = (long long)(*j - *_j) * _w % p;
	if(ntt_a != ntt_sup) std :: copy(a, a + n, ntt_a);
}
int _ntt_sup[N];
inline void ntt(auto a[], auto b[], int n, const int g) {
	auto ntt_a =  ntt_sup, j_a = a, _j_a = a;
	auto ntt_b = _ntt_sup, j_b = b, _j_b = b; int i, k, _k, w, _w;
	for(i = 1, w = pow_modp(g, (p - 1) / n); i < n; i <<= 1, w = (long long)w * w % p, std :: swap(a, ntt_a), std :: swap(b, ntt_b))
		for(k = 0, j_a = a, j_b = b, _j_a = a + (n >> 1), _j_b = b + (n >> 1), _w = 1; k != n; k += i, _w = (long long)_w * w % p)
			for(_k = k, k += i; _k != k; ++j_a, ++j_b, ++_j_a, ++_j_b, ++_k)
				ntt_a[_k] = *j_a + *_j_a < 0 ? *j_a + *_j_a + p : *j_a + *_j_a - p,
				ntt_b[_k] = *j_b + *_j_b < 0 ? *j_b + *_j_b + p : *j_b + *_j_b - p,
				ntt_a[_k + i] = (long long)(*j_a - *_j_a) * _w % p,
				ntt_b[_k + i] = (long long)(*j_b - *_j_b) * _w % p;
	if(ntt_a != ntt_sup) std :: copy(a, a + n, ntt_a),
	                     std :: copy(b, b + n, ntt_b);
}
inline void poly_mul(auto a[], auto b[], int n) {
	ntt(a, b, n, g);     for(int i = 0; i < n; ++i) a[i] = (long long)a[i] * b[i] % p; ntt(a, n, h);
	int inv_n = pow_modp(n, p - 2); for(int i = 0; i < n; ++i) a[i] = (long long)a[i] * inv_n % p;
	for(int i = 0; i < n; ++i) a[i] = a[i] < 0 ? a[i] + p : a[i];
}

int n, n_a, n_b;
int a[N], b[N];

int main(void) {
	fastIO :: read(n_a), fastIO :: read(n_b); for(n = 1; n <= n_a + n_b; n <<= 1);
	for(int i = 0; i <= n_a; ++i) fastIO :: read(a[i]);
	for(int i = 0; i <= n_b; ++i) fastIO :: read(b[i]);
	poly_mul(a, b, n);
	fastIO :: write(a, n_a + n_b + 1), fastIO :: output();
	return 0;
}

CompilationN/AN/ACompile OKScore: N/A

Subtask #1 Testcase #17.73 us36 KBAcceptedScore: 0

Subtask #1 Testcase #220.287 ms7 MB + 256 KBAcceptedScore: 100

Subtask #1 Testcase #38.582 ms2 MB + 792 KBAcceptedScore: 0

Subtask #1 Testcase #48.62 ms2 MB + 772 KBAcceptedScore: 0

Subtask #1 Testcase #59.93 us36 KBAcceptedScore: 0

Subtask #1 Testcase #67.68 us36 KBAcceptedScore: 0

Subtask #1 Testcase #78.31 us36 KBAcceptedScore: 0

Subtask #1 Testcase #819.47 ms6 MB + 680 KBAcceptedScore: 0

Subtask #1 Testcase #919.479 ms6 MB + 680 KBAcceptedScore: 0

Subtask #1 Testcase #1018.637 ms6 MB + 76 KBAcceptedScore: 0

Subtask #1 Testcase #1120.523 ms7 MB + 420 KBAcceptedScore: 0

Subtask #1 Testcase #1217.337 ms5 MB + 176 KBAcceptedScore: 0

Subtask #1 Testcase #137.06 us32 KBAcceptedScore: 0


Judge Duck Online | 评测鸭在线
Server Time: 2026-03-18 14:58:15 | Loaded in 1 ms | Server Status
个人娱乐项目,仅供学习交流使用 | 捐赠