結果

問題 No.2031 Colored Brackets
ユーザー norioc
提出日時 2025-03-25 07:01:04
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,790 bytes
コンパイル時間 297 ms
コンパイル使用メモリ 82,364 KB
実行使用メモリ 335,252 KB
最終ジャッジ日時 2025-03-25 07:01:08
合計ジャッジ時間 4,442 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other TLE * 1 -- * 29
権限があれば一括ダウンロードができます

ソースコード

diff #

from itertools import accumulate

MOD = 998244353
N = int(input())
S = input()

lparens = []
for c in S:
    v = 1 if c == '(' else -1
    lparens.append(v)

acc = list(accumulate(lparens))

ns = len(S)
dp = [[-1] * (ns+1) for _ in range(ns+1)]
dp[0][0] = 1
for i, c in enumerate(S):
    pp = [[0] * (ns+1) for _ in range(ns+1)]
    dp, pp = pp, dp

    lpar = 0
    if i > 0:
        lpar = acc[i-1]  # 開きカッコの個数

    if c == '(':
        # 赤を増やす
        for j in range(ns):
            k = lpar - j
            dp[j+1][k] += pp[j][k]
            dp[j+1][k] %= MOD
            # for k in range(ns):
            #     dp[j+1][k] += pp[j][k]
            #     dp[j+1][k] %= MOD

        for k in range(ns):
            j = lpar - k
            dp[j][k+1] += pp[j][k]
            dp[j][k+1] %= MOD

        # # 青を増やす
        # for j in range(ns):
        #     k = lpar - j
        #     dp[j][k+1] += pp[j][k]
        #     dp[j][k+1] %= MOD
        #     # for k in range(ns):
        #     #     dp[j][k+1] += pp[j][k]
        #     #     dp[j][k+1] %= MOD
    elif c == ')':
        # 赤を減らす
        for j in range(ns):
            k = lpar - j
            if j == 0: continue
            dp[j-1][k] += pp[j][k]
            dp[j-1][k] %= MOD
            # for k in range(ns):
            #     if j == 0: continue
            #     dp[j-1][k] += pp[j][k]
            #     dp[j-1][k] %= MOD

        # 青を減らす
        for j in range(ns):
            k = lpar - j
            if k < 0: continue
            dp[j][k-1] += pp[j][k]
            dp[j][k-1] %= MOD

            # for k in range(ns):
            #     if k == 0: continue
            #     dp[j][k-1] += pp[j][k]
            #     dp[j][k-1] %= MOD

ans = dp[0][0]
print(ans)
0