結果

問題 No.2106 Wild Cacco
ユーザー gew1fw
提出日時 2025-06-12 14:44:05
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 2,220 bytes
コンパイル時間 175 ms
コンパイル使用メモリ 82,968 KB
実行使用メモリ 77,340 KB
最終ジャッジ日時 2025-06-12 14:44:56
合計ジャッジ時間 4,870 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 5
other AC * 7 TLE * 1 -- * 22
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
from collections import defaultdict

MOD = 998244353

def main():
    s = sys.stdin.readline().strip()
    dp = defaultdict(int)
    dp[(0, 0)] = 1

    for c in s:
        next_dp = defaultdict(int)
        for (min_prev, max_prev), count in dp.items():
            if c == '(':
                new_min = min_prev + 1
                new_max = max_prev + 1
                if new_max < 0:
                    continue
                new_min = max(new_min, 0)
                if new_min <= new_max:
                    next_dp[(new_min, new_max)] = (next_dp[(new_min, new_max)] + count) % MOD
            elif c == ')':
                new_min = min_prev - 1
                new_max = max_prev - 1
                if new_max < 0:
                    continue
                new_min = max(new_min, 0)
                if new_min <= new_max:
                    next_dp[(new_min, new_max)] = (next_dp[(new_min, new_max)] + count) % MOD
            elif c == '?':
                new_min = min_prev - 1
                new_max = max_prev + 1
                if new_max < 0:
                    continue
                new_min = max(new_min, 0)
                if new_min <= new_max:
                    next_dp[(new_min, new_max)] = (next_dp[(new_min, new_max)] + count) % MOD
            else:  # c is '.'
                for option in ['(', ')', '?']:
                    if option == '(':
                        nmin = min_prev + 1
                        nmax = max_prev + 1
                    elif option == ')':
                        nmin = min_prev - 1
                        nmax = max_prev - 1
                    else:  # '?'
                        nmin = min_prev - 1
                        nmax = max_prev + 1
                    if nmax < 0:
                        continue
                    nmin = max(nmin, 0)
                    if nmin > nmax:
                        continue
                    next_dp[(nmin, nmax)] = (next_dp[(nmin, nmax)] + count) % MOD
        dp = next_dp

    total = 0
    for (min_val, max_val), count in dp.items():
        if min_val <= 0 <= max_val:
            total = (total + count) % MOD
    print(total % MOD)

if __name__ == '__main__':
    main()
0