結果
問題 | No.1068 #いろいろな色 / Red and Blue and more various colors (Hard) |
ユーザー | yuppe19 😺 |
提出日時 | 2020-06-12 15:53:51 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 2,962 ms / 3,500 ms |
コード長 | 2,187 bytes |
コンパイル時間 | 917 ms |
コンパイル使用メモリ | 77,192 KB |
実行使用メモリ | 15,912 KB |
最終ジャッジ日時 | 2023-09-06 09:17:10 |
合計ジャッジ時間 | 56,729 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
4,380 KB |
testcase_01 | AC | 1 ms
4,380 KB |
testcase_02 | AC | 1 ms
4,376 KB |
testcase_03 | AC | 43 ms
4,376 KB |
testcase_04 | AC | 29 ms
4,376 KB |
testcase_05 | AC | 32 ms
4,376 KB |
testcase_06 | AC | 25 ms
4,380 KB |
testcase_07 | AC | 23 ms
4,376 KB |
testcase_08 | AC | 31 ms
4,380 KB |
testcase_09 | AC | 35 ms
4,380 KB |
testcase_10 | AC | 14 ms
4,376 KB |
testcase_11 | AC | 22 ms
4,380 KB |
testcase_12 | AC | 12 ms
4,380 KB |
testcase_13 | AC | 2,946 ms
15,768 KB |
testcase_14 | AC | 2,940 ms
15,820 KB |
testcase_15 | AC | 2,962 ms
15,836 KB |
testcase_16 | AC | 2,938 ms
15,864 KB |
testcase_17 | AC | 2,934 ms
15,884 KB |
testcase_18 | AC | 2,937 ms
15,824 KB |
testcase_19 | AC | 2,940 ms
15,824 KB |
testcase_20 | AC | 2,937 ms
15,828 KB |
testcase_21 | AC | 2,934 ms
15,820 KB |
testcase_22 | AC | 2,941 ms
15,828 KB |
testcase_23 | AC | 2,945 ms
15,852 KB |
testcase_24 | AC | 2,937 ms
15,820 KB |
testcase_25 | AC | 2,951 ms
15,776 KB |
testcase_26 | AC | 2,939 ms
15,772 KB |
testcase_27 | AC | 2,941 ms
15,912 KB |
testcase_28 | AC | 2,937 ms
15,820 KB |
testcase_29 | AC | 2,924 ms
15,848 KB |
testcase_30 | AC | 2,949 ms
15,868 KB |
testcase_31 | AC | 1 ms
4,380 KB |
ソースコード
#include <cstdint> #include <cstdio> #include <queue> #include <map> #include <vector> using namespace std; using u32 = uint32_t; constexpr u32 O = 3, M = 998'244'353, M2 = M - 2; inline constexpr u32 mod_mul(uint64_t x, u32 y) { return static_cast<u32>(x * y % M); } map<pair<u32, u32>, u32> cache; inline u32 mod_pow(u32 a, u32 n) { auto key = make_pair(a, n); if(cache.count(key)) { return cache[key]; } u32 res = 1; for(; n; n>>=1) { if(n & 1) { res = mod_mul(res, a); } a = mod_mul(a, a); } return cache[key] = res; } void myfmt(vector<u32> &a, bool inv) { u32 n = static_cast<u32>(a.size()); if(n == 1) { return; } size_t m = n / 2; vector<u32> a0(m), a1(m); for(size_t i=0, j=0; i<m; ++i) { a0[i] = a[j++]; a1[i] = a[j++]; } myfmt(a0, inv); myfmt(a1, inv); u32 z = mod_pow(O, (M-1)/n); if(inv) { z = mod_pow(z, M2); } u32 pz = 1; for(size_t i=0, k=0; k<m; ++i, ++k) { a[k] = a0[i] + mod_mul(a1[i], pz); if(a[k] >= M) { a[k] -= M; } pz = mod_mul(pz, z); } for(size_t i=0, k=m; k<n; ++i, ++k) { a[k] = a0[i] + mod_mul(a1[i], pz); if(a[k] >= M) { a[k] -= M; } pz = mod_mul(pz, z); } } void fmt(vector<u32> &a) { myfmt(a, false); } void ifmt(vector<u32> &a) { myfmt(a, true); u32 n = static_cast<u32>(a.size()); u32 inv = mod_pow(n, M2); for(size_t i=0; i<n; ++i) { a[i] = mod_mul(a[i], inv); } } vector<u32> convol(vector<u32> &a, vector<u32> &b) { size_t n = 1; while(n < a.size() + b.size()) { n <<= 1; } a.resize(n); b.resize(n); fmt(a); fmt(b); vector<u32> c(n); for(size_t i=0; i<n; ++i) { c[i] = mod_mul(a[i], b[i]); } ifmt(c); return c; } int main(void) { int N, Q; scanf("%d%d", &N, &Q); queue<vector<u32>> que; for(int i=0; i<N; ++i) { uint64_t a; scanf("%lu", &a); que.push({static_cast<u32>(--a % M), 1}); } while(que.size() > 1) { vector<u32> a = que.front(); que.pop(); vector<u32> b = que.front(); que.pop(); que.emplace(convol(a, b)); } vector<u32> v = que.front(); for(int q=0; q<Q; ++q) { int pos; scanf("%d", &pos); printf("%u\n", v[pos]); } return 0; }