結果

問題 No.2219 Re:010
ユーザー H20H20
提出日時 2023-02-17 23:43:51
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 275 ms / 2,000 ms
コード長 682 bytes
コンパイル時間 206 ms
コンパイル使用メモリ 82,632 KB
実行使用メモリ 115,456 KB
最終ジャッジ日時 2024-07-19 14:53:02
合計ジャッジ時間 4,781 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
52,480 KB
testcase_01 AC 38 ms
52,096 KB
testcase_02 AC 35 ms
52,096 KB
testcase_03 AC 194 ms
102,916 KB
testcase_04 AC 74 ms
74,880 KB
testcase_05 AC 166 ms
94,592 KB
testcase_06 AC 90 ms
81,408 KB
testcase_07 AC 125 ms
91,264 KB
testcase_08 AC 188 ms
96,832 KB
testcase_09 AC 179 ms
96,308 KB
testcase_10 AC 159 ms
98,432 KB
testcase_11 AC 141 ms
93,952 KB
testcase_12 AC 139 ms
94,336 KB
testcase_13 AC 222 ms
113,816 KB
testcase_14 AC 221 ms
114,308 KB
testcase_15 AC 224 ms
113,940 KB
testcase_16 AC 219 ms
114,200 KB
testcase_17 AC 220 ms
114,160 KB
testcase_18 AC 92 ms
115,456 KB
testcase_19 AC 93 ms
115,200 KB
testcase_20 AC 275 ms
115,188 KB
testcase_21 AC 39 ms
51,584 KB
testcase_22 AC 37 ms
52,096 KB
testcase_23 AC 37 ms
51,968 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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