結果
問題 | No.2720 Sum of Subarray of Subsequence of... |
ユーザー | ecottea |
提出日時 | 2024-01-26 04:07:25 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 316 ms / 4,000 ms |
コード長 | 2,298 bytes |
コンパイル時間 | 5,034 ms |
コンパイル使用メモリ | 279,224 KB |
実行使用メモリ | 34,368 KB |
最終ジャッジ日時 | 2024-04-05 20:50:32 |
合計ジャッジ時間 | 8,958 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,120 KB |
testcase_01 | AC | 2 ms
5,120 KB |
testcase_02 | AC | 2 ms
5,120 KB |
testcase_03 | AC | 2 ms
5,120 KB |
testcase_04 | AC | 1 ms
5,120 KB |
testcase_05 | AC | 1 ms
5,120 KB |
testcase_06 | AC | 1 ms
5,120 KB |
testcase_07 | AC | 2 ms
5,120 KB |
testcase_08 | AC | 1 ms
5,120 KB |
testcase_09 | AC | 2 ms
5,120 KB |
testcase_10 | AC | 2 ms
5,120 KB |
testcase_11 | AC | 2 ms
5,120 KB |
testcase_12 | AC | 1 ms
5,120 KB |
testcase_13 | AC | 2 ms
5,120 KB |
testcase_14 | AC | 99 ms
9,196 KB |
testcase_15 | AC | 116 ms
11,328 KB |
testcase_16 | AC | 131 ms
13,044 KB |
testcase_17 | AC | 142 ms
14,292 KB |
testcase_18 | AC | 149 ms
15,524 KB |
testcase_19 | AC | 171 ms
17,652 KB |
testcase_20 | AC | 179 ms
18,356 KB |
testcase_21 | AC | 187 ms
18,756 KB |
testcase_22 | AC | 199 ms
19,352 KB |
testcase_23 | AC | 198 ms
20,432 KB |
testcase_24 | AC | 199 ms
23,488 KB |
testcase_25 | AC | 1 ms
5,120 KB |
testcase_26 | AC | 2 ms
5,120 KB |
testcase_27 | AC | 70 ms
12,552 KB |
testcase_28 | AC | 95 ms
8,676 KB |
testcase_29 | AC | 94 ms
8,676 KB |
testcase_30 | AC | 316 ms
34,368 KB |
testcase_31 | AC | 301 ms
33,356 KB |
testcase_32 | AC | 153 ms
17,552 KB |
ソースコード
#define _CRT_SECURE_NO_WARNINGS #include <bits/stdc++.h> using namespace std; #include <atcoder/all> using namespace atcoder; using mint = modint998244353; int main() { int n, m; cin >> n >> m; assert(1 <= n && n <= (int)1e5); assert(1 <= m && m <= (int)1e5); vector<int> a(n); for (int i = 0; i < n; i++) { cin >> a[i]; assert(0 <= a[i] && a[i] < 998244353); } string s; cin >> s; for (int j = 0; j < m; j++) { assert(s[j] == 'a' || s[j] == 's'); } // g_j(x) を (1-x)^(e_1) (1-2x)^(e_2) (1-3x)^(e_3) ... と表したときの指数列を更新していく. deque<int> q{ -1 }; int q_sum = -1; for(auto c : s) { if (c == 's') { q.push_front(-q_sum - 1); q_sum = -1; } else if (c == 'a') { q.front()--; q_sum--; } } q.push_front(0); int K = (int)q.size(); // g_M(x) の (1-kx) たちを分子と分母に振り分ける. vector<vector<mint>> nums{ {1} }, dnms{ {1} }; for (int k = 1; k < K; k++) { if (q[k] > 0) { for (int tmp = 0; tmp < q[k]; tmp++) { nums.push_back(vector<mint>{1, -k}); } } else if (q[k] < 0) { for (int tmp = 0; tmp < -q[k]; tmp++) { dnms.push_back(vector<mint>{1, -k}); } } } // 分子を分割統治法で求める. int Dnum = (int)nums.size(); for (int d = 1; d < Dnum; d *= 2) { for (int i = 0; i + d < Dnum; i += 2 * d) { nums[i] = convolution(nums[i], nums[i + d]); } } // 分母を分割統治法で求める. int Ddnm = (int)dnms.size(); for (int d = 1; d < Ddnm; d *= 2) { for (int i = 0; i + d < Ddnm; i += 2 * d) { dnms[i] = convolution(dnms[i], dnms[i + d]); } } // 分母の形式的冪級数としての逆元を求める. dnms[0].resize(n); vector<mint> dnm_inv{ dnms[0][0].inv() }; for (int k = 1; k < n; k *= 2) { int len = min(2 * k, n); vector<mint> tmp(len); int i_ub = min(len, n); for (int i = 0; i < i_ub; i++) tmp[i] = -dnms[0][i]; tmp = convolution(tmp, dnm_inv); tmp.resize(len); tmp[0] += 2; dnm_inv = convolution(dnm_inv, tmp); dnm_inv.resize(len); } // g_M(x) を求める. auto f = convolution(nums[0], dnm_inv); // 答えへの寄与を足し合わせる. mint res; for (int i = 0; i < n; i++) { int l = i; int r = n - 1 - i; res += a[i] * f[l] * f[r]; } cout << res.val() << endl; }