結果
問題 | No.931 Multiplicative Convolution |
ユーザー | msm1993 |
提出日時 | 2020-10-29 17:40:23 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 122 ms / 2,000 ms |
コード長 | 2,737 bytes |
コンパイル時間 | 443 ms |
コンパイル使用メモリ | 53,032 KB |
実行使用メモリ | 11,392 KB |
最終ジャッジ日時 | 2024-07-21 22:36:06 |
合計ジャッジ時間 | 3,258 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,816 KB |
testcase_01 | AC | 1 ms
6,940 KB |
testcase_02 | AC | 2 ms
6,944 KB |
testcase_03 | AC | 1 ms
6,940 KB |
testcase_04 | AC | 1 ms
6,940 KB |
testcase_05 | AC | 2 ms
6,944 KB |
testcase_06 | AC | 3 ms
6,940 KB |
testcase_07 | AC | 12 ms
6,940 KB |
testcase_08 | AC | 99 ms
11,264 KB |
testcase_09 | AC | 86 ms
11,392 KB |
testcase_10 | AC | 96 ms
11,264 KB |
testcase_11 | AC | 87 ms
11,264 KB |
testcase_12 | AC | 54 ms
7,552 KB |
testcase_13 | AC | 122 ms
11,324 KB |
testcase_14 | AC | 104 ms
11,392 KB |
testcase_15 | AC | 96 ms
11,264 KB |
testcase_16 | AC | 95 ms
11,392 KB |
コンパイルメッセージ
main.cpp: In function 'int main()': main.cpp:82:56: warning: 'x' may be used uninitialized [-Wmaybe-uninitialized] 82 | for (int i = 1; i < n; i++) p[i] = (p[i - 1] * x) % pn; | ^ main.cpp:63:24: note: 'x' was declared here 63 | int n, sz = 1, x, pn; | ^
ソースコード
bool debug = false; #include <stdio.h> #include <vector> using namespace std; constexpr int kMod = 998244353, kN = int(1E5 + 10); typedef long long int ll; ll Pow(ll a, ll b) { ll ans = 1; while (b) { if (b & 1) ans = (ans * a) % kMod; b >>= 1; a = (a * a) % kMod; } return ans; } ll Rev(ll n) {return Pow(n, kMod - 2);} void Ntt(vector<ll> &v, bool on, int sz) { ll wn, w, u, t, inv; for (int i = 1, j = sz >> 1, k; i < (sz - 1); i++) { if (i < j) swap(v[i], v[j]); k = sz >> 1; while (j & k) { j ^= k; k >>= 1; } j |= k; } for (int i = 2; i <= sz; i <<= 1) { wn = on ? Pow(3, (kMod - 1) / i) : Pow(3, kMod - 1 - (kMod - 1) / i); for (int j = 0; j < sz; j += i) { w = 1; for (int k = j; k < j + (i >> 1); k++) { u = v[k]; t = (w * v[k + (i >> 1)]) % kMod; v[k] = (u + t) % kMod; v[k + (i >> 1)] = (u - t + kMod) % kMod; w = (w * wn) % kMod; } } } if (on) { inv = Rev(sz); for (int i = 0; i < sz; i++) v[i] = (v[i] * inv) % kMod; } return ; } bool ok(int n, int pn) { int cnt = 2; ll now = n * n % pn; while (now != n) { now = (now * n) % pn; cnt++; } return cnt == pn; } ll p[kN], rp[kN]; int a[kN], b[kN]; int main() { int n, sz = 1, x, pn; vector<ll> va, vb, vc; scanf("%d", &n); pn = n; n--; for (int i = 0; i < n; i++) scanf("%d", &a[i]); for (int i = 0; i < n; i++) scanf("%d", &b[i]); while (sz < n) sz <<= 1; sz <<= 1; if (debug) printf("sz = %d\n", sz); va.resize(sz); vb.resize(sz); vc.resize(sz); for (int i = 2; i < pn; i++) if (ok(i, pn)) { x = i; break; } if (debug) printf("x = %d\n", x); p[0] = 1; for (int i = 1; i < n; i++) p[i] = (p[i - 1] * x) % pn; for (int i = 0; i < n; i++) p[i]--; for (int i = 0; i < n; i++) rp[p[i]] = i; if (debug) { printf("p::"); for (int i = 0; i < n; i++) printf(" %lld", p[i]); printf("\n"); printf("rp::"); for (int i = 0; i < n; i++) printf(" %lld", rp[i]); printf("\n"); } for (int i = 0; i < n; i++) va[i] = a[p[i]]; for (int i = 0; i < n; i++) vb[i] = b[p[i]]; if (debug) { printf("va::"); for (int i = 0; i < sz; i++) printf(" %lld", va[i]); printf("\n"); } if (debug) { printf("vb::"); for (int i = 0; i < sz; i++) printf(" %lld", vb[i]); printf("\n"); } Ntt(va, false, sz); Ntt(vb, false, sz); for (int i = 0; i < sz; i++) vc[i] = (va[i] * vb[i]) % kMod; Ntt(vc, true, sz); if (debug) { printf("vc::"); for (int i = 0; i < sz; i++) printf(" %lld", vc[i]); printf("\n"); } for (int i = n; i < sz; i++) vc[i % n] += vc[i]; for (int i = 0; i < n; i++) vc[i] %= kMod; for (int i = 0; i < n; i++) if (vc[i] < 0) vc[i] += kMod; printf("%lld", vc[0]); for (int i = 1; i < n; i++) printf(" %lld", vc[rp[i]]); printf("\n"); }