結果
問題 |
No.2541 Divide 01 String
|
ユーザー |
|
提出日時 | 2023-12-10 17:12:33 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 682 bytes |
コンパイル時間 | 185 ms |
コンパイル使用メモリ | 82,284 KB |
実行使用メモリ | 89,604 KB |
最終ジャッジ日時 | 2024-09-27 04:08:19 |
合計ジャッジ時間 | 2,230 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 11 WA * 9 |
ソースコード
def RLE(S: list) -> list: """ 入力をランレングス圧縮したリストを返す. [(value1,length1),(value2,length2),...] Parameters ---------- S:list Examples -------- >>> RLE('aaabbc') [('a', 3), ('b', 2), ('c', 1)] """ from itertools import groupby res = [(k, len(list(g))) for k, g in groupby(S)] return res N = int(input()) S = input() R = RLE(S) ans = 1 MOD = 998244353 for i in range(1, len(R) - 1): l = R[i - 1][0] mid = R[i][0] r = R[i + 1][0] if l == r == "1" and mid == "0": ans *= R[i][1] + 2 if mid == "1": ans *= pow(2, R[i][1] - 1, MOD) ans %= MOD print(ans)