結果
問題 | No.2720 Sum of Subarray of Subsequence of... |
ユーザー |
![]() |
提出日時 | 2024-04-05 23:33:43 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 278 ms / 4,000 ms |
コード長 | 2,546 bytes |
コンパイル時間 | 4,551 ms |
コンパイル使用メモリ | 293,000 KB |
実行使用メモリ | 17,992 KB |
最終ジャッジ日時 | 2024-10-01 03:09:48 |
合計ジャッジ時間 | 7,951 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 31 |
ソースコード
#include<bits/stdc++.h>namespace {#pragma GCC diagnostic ignored "-Wunused-function"#include<atcoder/all>#pragma GCC diagnostic warning "-Wunused-function"using namespace std;using namespace atcoder;#define rep(i,n) for(int i = 0; i < (int)(n); i++)#define rrep(i,n) for(int i = (int)(n) - 1; i >= 0; i--)#define all(x) begin(x), end(x)#define rall(x) rbegin(x), rend(x)template<class T> bool chmax(T& a, const T& b) { if (a < b) { a = b; return true; } else return false; }template<class T> bool chmin(T& a, const T& b) { if (b < a) { a = b; return true; } else return false; }using ll = long long;using P = pair<int,int>;using VI = vector<int>;using VVI = vector<VI>;using VL = vector<ll>;using VVL = vector<VL>;using mint = modint998244353;vector<mint> fps_inverse(vector<mint> f, int precision) {int n = f.size();assert(n >= 1 && f[0] != 0);int len = 1;const int z = bit_ceil(1U * precision);vector<mint> g{f[0].inv()};const mint inv4 = mint(4).inv();mint inv4k = 1;while (len < z) {int nlen = 2 * len;vector<mint> ft(f.begin(), f.begin() + min(n, nlen));ft.resize(nlen);butterfly(ft);vector<mint> gt = g;gt.resize(nlen);internal::butterfly(gt);for (int i = 0; i < nlen; i++) ft[i] *= gt[i];internal::butterfly_inv(ft);ft.erase(ft.begin(), ft.begin() + len);ft.resize(nlen);internal::butterfly(ft);for (int i = 0; i < nlen; i++) ft[i] *= gt[i];internal::butterfly_inv(ft);inv4k *= inv4;mint c = -inv4k;for (int i = 0; i < len; i++) g.emplace_back(c * ft[i]);len = nlen;}g.resize(precision);return g;}} int main() {ios::sync_with_stdio(false);cin.tie(0);int n, m;cin >> n >> m;VI a(n);rep(i, n) cin >> a[i];string s;cin >> s;VI cnt(m + 1);rrep(i, m) cnt[i] = cnt[i+1] + (s[i] == 's');using VM = vector<mint>;deque<VM> q1{{1}}, q2{{1}};q2.emplace_back(VM{1, -(cnt[0] + 1)});rep(i, m) if (s[i] == 'a') {int si = cnt[i+1] + 1;q1.emplace_back(VM{1, -si + 1});q2.emplace_back(VM{1, -si});}rep(_, 2) {while (q1.size() >= 2) {auto f = move(q1.front()); q1.pop_front();auto g = move(q1.front()); q1.pop_front();q1.emplace_back(convolution(f, g));}swap(q1, q2);}auto f = move(q1.front()); q1.pop_front();auto g = move(q2.front()); q2.pop_front();f = convolution(fps_inverse(g, n + 1), f);mint ans;rep(i, n) {ans += a[i] * f[i] * f[n - (i + 1)];}cout << ans.val() << '\n';}