結果
問題 | No.2219 Re:010 |
ユーザー |
![]() |
提出日時 | 2022-01-26 09:01:55 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 14 ms / 2,000 ms |
コード長 | 713 bytes |
コンパイル時間 | 2,543 ms |
コンパイル使用メモリ | 198,544 KB |
最終ジャッジ日時 | 2025-01-27 15:20:11 |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 21 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/modint> using namespace std; using mint = atcoder::modint998244353; int main() { string s; cin >> s; int n = s.size(); vector<int> csum0(n + 1), csumq(n + 1); for (int i = 0; i < n; i++) { csum0[i + 1] = csum0[i] + (s[i] == '0'); csumq[i + 1] = csumq[i] + (s[i] == '?'); } mint ans = 0; const mint inv2 = mint(2).inv(); for (int i = 0; i < n; i++) { mint L = csum0[i] + csumq[i] * inv2; mint R = (csum0[n] - csum0[i + 1]) + (csumq[n] - csumq[i + 1]) * inv2; mint E = L * R; if (s[i] == '0') E = 0; else if (s[i] == '?') E *= inv2; ans += E; } ans *= mint(2).pow(csumq[n]); cout << ans.val() << "\n"; }