結果
問題 | No.2219 Re:010 |
ユーザー |
![]() |
提出日時 | 2023-10-31 17:29:35 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 75 ms / 2,000 ms |
コード長 | 1,235 bytes |
コンパイル時間 | 467 ms |
コンパイル使用メモリ | 82,460 KB |
実行使用メモリ | 79,116 KB |
最終ジャッジ日時 | 2024-09-25 17:41:43 |
合計ジャッジ時間 | 2,858 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 21 |
ソースコード
# 0,?をそれぞれカウント DIV = 998244353 S = input() N = len(S) zero = [0] * N ques = [0] * N if S[0] == "0": zero[0] = 1 if S[0] == "?": ques[0] = 1 for i in range(1, N): zero[i] = zero[i - 1] + (S[i] == "0") ques[i] = ques[i - 1] + (S[i] == "?") ans = 0 allq = ques[-1] allpat = pow(2, allq, DIV) minuspat = pow(2, allq - 1, DIV) if allq >= 1 else 0 minus2pat = pow(2, allq - 2, DIV) if allq >= 2 else 0 minus3pat = pow(2, allq - 3, DIV) if allq >= 3 else 0 for i in range(1, N - 1): if S[i] == "1": # 既存の0を使う場合、?の使い方は全パターンある ans += zero[i - 1] * (zero[-1] - zero[i]) * allpat # 0と?を使う場合。 ans += zero[i - 1] * (ques[-1] - ques[i]) * minuspat # ?と0 ans += ques[i - 1] * (zero[-1] - zero[i]) * minuspat # ?と? ans += ques[i - 1] * (ques[-1] - ques[i]) * minus2pat ans %= DIV elif S[i] == "?": ans += zero[i - 1] * (zero[-1] - zero[i]) * minuspat ans += zero[i - 1] * (ques[-1] - ques[i]) * minus2pat ans += ques[i - 1] * (zero[-1] - zero[i]) * minus2pat ans += ques[i - 1] * (ques[-1] - ques[i]) * minus3pat ans %= DIV print(ans)