結果

問題 No.2219 Re:010
ユーザー roarisroaris
提出日時 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
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
51,840 KB
testcase_01 AC 42 ms
51,840 KB
testcase_02 AC 40 ms
51,840 KB
testcase_03 AC 109 ms
81,024 KB
testcase_04 AC 82 ms
75,264 KB
testcase_05 AC 104 ms
79,232 KB
testcase_06 AC 98 ms
77,312 KB
testcase_07 AC 103 ms
77,824 KB
testcase_08 AC 113 ms
80,000 KB
testcase_09 AC 115 ms
80,512 KB
testcase_10 AC 106 ms
78,976 KB
testcase_11 AC 98 ms
78,976 KB
testcase_12 AC 98 ms
78,848 KB
testcase_13 AC 106 ms
81,536 KB
testcase_14 AC 109 ms
81,280 KB
testcase_15 AC 106 ms
81,280 KB
testcase_16 AC 111 ms
81,408 KB
testcase_17 AC 114 ms
81,408 KB
testcase_18 AC 58 ms
64,640 KB
testcase_19 AC 93 ms
81,024 KB
testcase_20 AC 99 ms
81,152 KB
testcase_21 AC 38 ms
51,456 KB
testcase_22 AC 38 ms
51,968 KB
testcase_23 AC 38 ms
51,712 KB
権限があれば一括ダウンロードができます

ソースコード

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