結果
| 問題 |
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 |
ソースコード
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)
norioc