結果
問題 | No.2219 Re:010 |
ユーザー |
|
提出日時 | 2023-02-17 22:04:45 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 455 ms / 2,000 ms |
コード長 | 1,441 bytes |
コンパイル時間 | 169 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 84,096 KB |
最終ジャッジ日時 | 2024-07-19 13:16:18 |
合計ジャッジ時間 | 5,401 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 21 |
ソースコード
import sys input = lambda :sys.stdin.readline()[:-1] ni = lambda :int(input()) na = lambda :list(map(int,input().split())) yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES") no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO") ####################################################################### s = input() n = len(s) mod = 998244353 lc = [0] * (1+n) rc = [0] * (1+n) for i in range(n): if s[i] == "0": lc[i+1] += lc[i] + 1 else: lc[i+1] = lc[i] s = s[::-1] for i in range(n): if s[i] == "0": rc[i+1] += rc[i] + 1 else: rc[i+1] = rc[i] s = s[::-1] ld = [0] * (1+n) rd = [0] * (1+n) for i in range(n): if s[i] == "?": ld[i+1] += ld[i] + 1 else: ld[i+1] = ld[i] s = s[::-1] for i in range(n): if s[i] == "?": rd[i+1] += rd[i] + 1 else: rd[i+1] = rd[i] s = s[::-1] ans = 0 x = s.count("?") for i in range(n): if s[i] == "1": ans += lc[i] * rc[n-1-i] * pow(2, x, mod) ans += ld[i] * rc[n-1-i] * pow(2, x-1, mod) ans += lc[i] * rd[n-1-i] * pow(2, x-1, mod) ans += ld[i] * rd[n-1-i] * pow(2, x-2, mod) if s[i] == "?": ans += lc[i] * rc[n-1-i] * pow(2, x-1, mod) ans += ld[i] * rc[n-1-i] * pow(2, x-2, mod) ans += lc[i] * rd[n-1-i] * pow(2, x-2, mod) ans += ld[i] * rd[n-1-i] * pow(2, x-3, mod) ans %= mod print(ans)