結果
問題 | No.1068 #いろいろな色 / Red and Blue and more various colors (Hard) |
ユーザー | yuppe19 😺 |
提出日時 | 2020-09-15 10:59:50 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 2,504 ms / 3,500 ms |
コード長 | 2,489 bytes |
コンパイル時間 | 2,178 ms |
コンパイル使用メモリ | 68,260 KB |
実行使用メモリ | 15,900 KB |
最終ジャッジ日時 | 2023-09-04 01:38:29 |
合計ジャッジ時間 | 48,180 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
4,376 KB |
testcase_01 | AC | 1 ms
4,380 KB |
testcase_02 | AC | 2 ms
4,376 KB |
testcase_03 | AC | 36 ms
4,376 KB |
testcase_04 | AC | 24 ms
4,376 KB |
testcase_05 | AC | 28 ms
4,376 KB |
testcase_06 | AC | 21 ms
4,376 KB |
testcase_07 | AC | 19 ms
4,376 KB |
testcase_08 | AC | 26 ms
4,376 KB |
testcase_09 | AC | 29 ms
4,376 KB |
testcase_10 | AC | 12 ms
4,376 KB |
testcase_11 | AC | 19 ms
4,380 KB |
testcase_12 | AC | 11 ms
4,376 KB |
testcase_13 | AC | 2,504 ms
15,832 KB |
testcase_14 | AC | 2,416 ms
15,864 KB |
testcase_15 | AC | 2,413 ms
15,872 KB |
testcase_16 | AC | 2,409 ms
15,804 KB |
testcase_17 | AC | 2,416 ms
15,772 KB |
testcase_18 | AC | 2,413 ms
15,900 KB |
testcase_19 | AC | 2,407 ms
15,836 KB |
testcase_20 | AC | 2,409 ms
15,844 KB |
testcase_21 | AC | 2,410 ms
15,836 KB |
testcase_22 | AC | 2,410 ms
15,776 KB |
testcase_23 | AC | 2,414 ms
15,796 KB |
testcase_24 | AC | 2,413 ms
15,836 KB |
testcase_25 | AC | 2,413 ms
15,824 KB |
testcase_26 | AC | 2,412 ms
15,832 KB |
testcase_27 | AC | 2,419 ms
15,864 KB |
testcase_28 | AC | 2,424 ms
15,824 KB |
testcase_29 | AC | 2,398 ms
15,888 KB |
testcase_30 | AC | 2,393 ms
15,796 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, D = 119, S = 23, M = D << S | 1, M2 = M - 2; inline constexpr u32 mod_mul(u32 x, uint64_t y) { return static_cast<u32>(x * y % M); } inline constexpr u32 mod_pow(u32 a, u32 n) { u32 res = 1; for(; n; n>>=1) { if(n & 1) { res = mod_mul(res, a); } a = mod_mul(a, a); } return res; } template <size_t N> struct Seq { u32 arr[N+N]; constexpr Seq() : arr() { arr[0] = mod_pow(O, D); arr[N] = mod_pow(arr[0], M-2); for(size_t i=0; i<N-1; ++i) { arr[ i+1] = mod_mul(arr[ i], arr[ i]); arr[N+i+1] = mod_mul(arr[N+i], arr[N+i]); } } }; constexpr auto cache = Seq<S>(); void myfmt(vector<u32> &a, bool inv, size_t ln) { 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, ln-1); myfmt(a1, inv, ln-1); u32 z = cache.arr[inv*S + S-ln]; 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, size_t ln) { myfmt(a, false, ln); } void ifmt(vector<u32> &a, size_t ln) { myfmt(a, true, ln); 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, ln = 0; while(n < a.size() + b.size() - 1) { n <<= 1; ++ln; } a.resize(n); b.resize(n); fmt(a, ln); fmt(b, ln); vector<u32> c(n); for(size_t i=0; i<n; ++i) { c[i] = mod_mul(a[i], b[i]); } ifmt(c, ln); 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; }