結果
問題 | No.22 括弧の対応 |
ユーザー |
|
提出日時 | 2023-04-21 15:48:57 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 41 ms / 5,000 ms |
コード長 | 1,184 bytes |
コンパイル時間 | 363 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 12,160 KB |
最終ジャッジ日時 | 2024-11-06 10:38:17 |
合計ジャッジ時間 | 1,789 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 19 |
ソースコード
from typing import Tuple def validate_input(N: int, K: int, S: str) -> Tuple[int, int, str]: """ 引数のバリデーションを行う関数 """ assert 1 <= N <= 10000, f"Invalid N: {N}" assert 1 <= K <= N, f"Invalid K: {K}" assert all(c in {"(", ")"} for c in S), f"Invalid S: {S}" return N, K, S def find_matching_bracket(N: int, K: int, S: str) -> int: """ 指定された括弧に対応する括弧の場所を取得する関数 """ _, _, S = validate_input(N, K, S) stack = [] bracket_pairs = {} # key: "(" の位置, value: ")" の位置 # という辞書を作成する for i, char in enumerate(S): if char == "(": stack.append(i) else: bracket_pairs[stack.pop()] = i # K 番目が "(" の場合は、対応する")" の位置を返す # ")" の場合は、対応する "(" の位置を返す if S[K - 1] == ")": # 辞書の key と value を入れ替える bracket_pairs = {v: k for k, v in bracket_pairs.items()} return bracket_pairs[K - 1] + 1 N, K = map(int, input().split()) S = input() print(find_matching_bracket(N, K, S))