結果

問題 No.2219 Re:010
コンテスト
ユーザー roaris
提出日時 2023-02-18 02:47:36
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 115 ms / 2,000 ms
コード長 701 bytes
コンパイル時間 163 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 81,536 KB
最終ジャッジ日時 2024-07-19 17:04:03
合計ジャッジ時間 3,113 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

S = input()[:-1]
N = len(S)
acc0 = [0]*(N+1)
acc2 = [0]*(N+1)

for i in range(N):
    acc0[i+1] = acc0[i]+(1 if S[i]=='0' else 0)
    acc2[i+1] = acc2[i]+(1 if S[i]=='?' else 0)

ans = 0
MOD = 998244353
pow2 = [1]*(N+1)

for i in range(N):
    pow2[i+1] = pow2[i]*2%MOD

for i in range(N):
    if S[i]=='0':
        continue
    
    for f1 in range(2):
        for f2 in range(2):
            l = acc0[i] if f1==0 else acc2[i]
            r = acc0[-1]-acc0[i+1] if f2==0 else acc2[-1]-acc2[i+1]
            q = acc2[-1]-f1-f2-(1 if S[i]=='?' else 0)
            
            if q>=0:
                ans += l*r*pow2[q]%MOD
                ans %= MOD

print(ans)
0