結果
問題 | No.2219 Re:010 |
ユーザー |
![]() |
提出日時 | 2023-02-17 22:09:17 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 94 ms / 2,000 ms |
コード長 | 608 bytes |
コンパイル時間 | 216 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 80,492 KB |
最終ジャッジ日時 | 2024-07-19 13:19:56 |
合計ジャッジ時間 | 2,673 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 21 |
ソースコード
S=input() N=len(S) c=S.count("?") ''' 主客転倒 i<j<kが0,1,0になる ?の数:cnt 2^(c-cnt) ''' mod=998244353 ans=0 A=[0 for i in range(N)] inv2=pow(2,mod-2,mod) for i in range(N): if S[i]=="0": A[i]=1 elif S[i]=="?": A[i]=inv2 else: A[i]=0 X=[0 for i in range(N+1)] Y=[0 for i in range(N+1)] for i in range(N): X[i+1]=X[i]+A[i] for i in range(N-1,-1,-1): Y[i]=Y[i+1]+A[i] for i in range(N): if S[i]=="1": ans+=X[i]*Y[i+1] ans%=mod elif S[i]=="?": ans+=X[i]*Y[i+1]*inv2 ans%=mod ans*=pow(2,c,mod) ans%=mod print(ans)