結果

問題 No.2219 Re:010
コンテスト
ユーザー H20
提出日時 2023-02-17 23:43:51
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 135 ms / 2,000 ms
コード長 682 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 143 ms
コンパイル使用メモリ 85,092 KB
実行使用メモリ 173,740 KB
最終ジャッジ日時 2026-04-02 17:08:07
合計ジャッジ時間 2,635 ms
ジャッジサーバーID
（参考情報） 		judge1_1 / judge3_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import itertools
S = list(input())
Q = [0]*len(S)
Z = [0]*len(S)
mod = 998244353
for i,s in enumerate(S):
    if s=='0':
        Z[i]=1
    if s=='?':
        Q[i]=1
ZAC = list(itertools.accumulate(Z))
QAC = list(itertools.accumulate(Q))
ans = 0

def modinv(a):
    return pow(a, mod-2, mod)
div2 = modinv(2)

for i,s in enumerate(S):
    if i==0 or i==len(S)-1:
        continue
    if s=='1' or s=='?':
        leftZ = ZAC[i-1]
        leftQ = QAC[i-1]
        rightZ = ZAC[-1]-ZAC[i]
        rightQ = QAC[-1]-QAC[i]
        left = (pow(2,leftQ,mod)*(leftZ+leftQ*div2))%mod
        right = (pow(2,rightQ,mod)*(rightZ+rightQ*div2))%mod
        ans = (ans+left*right)%mod
print(ans)
0