結果

問題 No.2031 Colored Brackets
ユーザー norioc
提出日時 2025-03-25 07:10:21
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 2,098 bytes
コンパイル時間 299 ms
コンパイル使用メモリ 82,308 KB
実行使用メモリ 167,208 KB
最終ジャッジ日時 2025-03-25 07:10:30
合計ジャッジ時間 8,721 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 3 TLE * 2 -- * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

from itertools import accumulate
from collections import defaultdict

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 = defaultdict(int)
# dp[0][0] = 1
dp[0, 0] = 1
for i, c in enumerate(S):
    #print(f'{i=}')
    # pp = [[0] * (ns+1) for _ in range(ns+1)]
    pp = defaultdict(int)
    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

            dp[k, j+1] += pp[j, k]
            dp[k, j+1] %= 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:
                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

                dp[k, j-1] += pp[j, k]
                dp[k, j-1] %= 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