結果
問題 |
No.2031 Colored Brackets
|
ユーザー |
![]() |
提出日時 | 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 |
ソースコード
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)