結果

問題 No.2031 Colored Brackets
コンテスト
ユーザー Shirotsume
提出日時 2022-06-12 04:44:08
言語 PyPy3
(7.3.15)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,461 bytes
コンパイル時間 466 ms
コンパイル使用メモリ 81,944 KB
実行使用メモリ 332,508 KB
最終ジャッジ日時 2024-09-15 16:26:46
合計ジャッジ時間 4,043 ms
ジャッジサーバーID
（参考情報） 		judge6 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other TLE * 1 -- * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

def is_correct(s):
    stack = []
    for v in s:
        stack.append(v)
        if len(stack) >= 2 and stack[-2] == '(' and stack[-1] == ')':
            stack.pop()
            stack.pop()
    if not stack:
        return True
    else:
        return False


def solve(n, s):
    mod = 998244353
    if not is_correct(s):
        return 0
    open = [0] * n
    close = [0] * n

    for i in range(n):
        if s[i] == '(':
            open[i] += 1
        else:
            close[i] += 1
        open[i] += open[i - 1]
        close[i] += close[i - 1]

    dp = [[0] * (n + 1) for _ in range(n + 1)]
    dp[0][0] = 1

    for i in range(n):
        e = [[0] * (n + 1) for _ in range(n + 1)]
        if s[i] == '(':
            for j in range(i + 1):
                for k in range(j + 1):
                    e[j + 1][k] += dp[j][k]
                    e[j + 1][k] %= mod
                    e[j][k] += dp[j][k]
                    e[j][k] %= mod
        else:
            for j in range(i + 1):
                for k in range(j + 1):
                    if j >= k + 1:
                        e[j][k + 1] += dp[j][k]
                        e[j][k + 1] %= mod
                    if open[i] - j >= close[i] - k:
                        e[j][k] += dp[j][k]
                        e[j][k] %= mod
        dp = e
    ans = 0
    for i in range(n):
        ans += dp[i][i]
        ans %= mod
    return ans

n = int(input())

s = input()

print(solve(n, s))
0