結果

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

ソースコード

diff # 

n = int(input())
s = input()
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)] for _ in range(n + 1)]
    dp[0][0][0] = 1

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

print(solve(n, s))
0