結果
問題 | No.2019 Digits Filling for All Substrings |
ユーザー | Tatsuyaaaa |
提出日時 | 2022-07-22 23:09:01 |
言語 | C++17 (gcc 13.2.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 245 ms / 2,000 ms |
コード長 | 2,498 bytes |
コンパイル時間 | 2,623 ms |
コンパイル使用メモリ | 220,712 KB |
実行使用メモリ | 55,860 KB |
最終ジャッジ日時 | 2023-09-17 11:58:31 |
合計ジャッジ時間 | 7,009 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
4,380 KB |
testcase_01 | AC | 1 ms
4,376 KB |
testcase_02 | AC | 2 ms
4,376 KB |
testcase_03 | AC | 1 ms
4,376 KB |
testcase_04 | AC | 170 ms
40,544 KB |
testcase_05 | AC | 115 ms
28,396 KB |
testcase_06 | AC | 103 ms
25,476 KB |
testcase_07 | AC | 73 ms
17,496 KB |
testcase_08 | AC | 242 ms
55,396 KB |
testcase_09 | AC | 242 ms
54,636 KB |
testcase_10 | AC | 243 ms
55,404 KB |
testcase_11 | AC | 243 ms
55,324 KB |
testcase_12 | AC | 243 ms
55,240 KB |
testcase_13 | AC | 245 ms
55,860 KB |
testcase_14 | AC | 3 ms
4,376 KB |
testcase_15 | AC | 4 ms
4,376 KB |
testcase_16 | AC | 14 ms
5,568 KB |
testcase_17 | AC | 5 ms
4,376 KB |
testcase_18 | AC | 5 ms
4,380 KB |
testcase_19 | AC | 2 ms
4,376 KB |
testcase_20 | AC | 2 ms
4,380 KB |
testcase_21 | AC | 1 ms
4,376 KB |
testcase_22 | AC | 2 ms
4,376 KB |
testcase_23 | AC | 1 ms
4,380 KB |
testcase_24 | AC | 111 ms
27,152 KB |
testcase_25 | AC | 152 ms
33,972 KB |
testcase_26 | AC | 74 ms
18,072 KB |
testcase_27 | AC | 20 ms
6,620 KB |
testcase_28 | AC | 19 ms
6,668 KB |
testcase_29 | AC | 140 ms
31,428 KB |
testcase_30 | AC | 201 ms
48,912 KB |
testcase_31 | AC | 197 ms
47,732 KB |
testcase_32 | AC | 59 ms
15,644 KB |
testcase_33 | AC | 145 ms
32,556 KB |
ソースコード
#include <bits/stdc++.h> #include <atcoder/modint> using namespace std; using namespace atcoder; using mint = modint998244353; #define ll long long struct A { // 1-----, 1?----, 123---, ... vector<mint> left; // -----6, ----56, ---456, ... vector<mint> right; // 1?3456 vector<mint> all; }; vector<mint> merge(vector<mint> x, vector<mint> y) { int n = x.size(); vector<mint> z(n, 0); for (int i=0;i<n;i++) { for (int j=0;j<n;j++) { z[(i+j)%n] += x[i]*y[j]; } } return z; } vector<mint> operator+(vector<mint> lhs, vector<mint> rhs) { vector<mint> result = lhs; int n = min((int)result.size(), (int)rhs.size()); for (int i=0;i<n;i++) result[i] += rhs[i]; return result; } vector<mint> operator-(vector<mint> lhs, vector<mint> rhs) { vector<mint> result = lhs; int n = min((int)result.size(), (int)rhs.size()); for (int i=0;i<n;i++) result[i] -= rhs[i]; return result; } int main() { ios::sync_with_stdio(0); cin.tie(0); int N; string S; cin >> N >> S; mint ans = 0; vector<A> vec; for (int i=0;i<N;i++) { vector<mint> tmp_left(3, 0); vector<mint> tmp_right(3, 0); vector<mint> tmp_all(3, 0); if (S[i] == '?') { tmp_left[0] = 4; tmp_left[1] = 3; tmp_left[2] = 3; tmp_right[0] = 4; tmp_right[1] = 3; tmp_right[2] = 3; tmp_all[0] = 4; tmp_all[1] = 3; tmp_all[2] = 3; } else { tmp_left[(S[i]-'0')%3] = 1; tmp_right[(S[i]-'0')%3] = 1; tmp_all[(S[i]-'0')%3] = 1; } ans += tmp_left[0]; vec.push_back(A{tmp_left, tmp_right, tmp_all}); } while (true) { int n = vec.size(); vector<A> vec_nxt; for (int i=0;i+1<n;i+=2) { vector<mint> add_vec = merge(vec[i].right, vec[i+1].left); ans += add_vec[0]; vector<mint> left_nxt = vec[i].left + merge(vec[i].all, vec[i+1].left); vector<mint> right_nxt = vec[i+1].right + merge(vec[i+1].all, vec[i].right); vector<mint> all_nxt = merge(vec[i].all, vec[i+1].all); vec_nxt.push_back(A{left_nxt, right_nxt, all_nxt}); } if (n % 2 == 1) { vec_nxt.push_back(vec[n-1]); } vec = vec_nxt; if (n == 1) break; } cout << ans.val() << "\n"; return 0; }