[Baekjoon] 9012/괄호/Python/파이썬/자료구조/문자열/스택

·2024년 12월 4일
0

문제풀이

목록 보기
3/56
post-thumbnail

💡문제

괄호 문자열(Parenthesis String, PS)은 두 개의 괄호 기호인 ‘(’ 와 ‘)’ 만으로 구성되어 있는 문자열이다. 그 중에서 괄호의 모양이 바르게 구성된 문자열을 올바른 괄호 문자열(Valid PS, VPS)이라고 부른다. 한 쌍의 괄호 기호로 된 “( )” 문자열은 기본 VPS 이라고 부른다. 만일 x 가 VPS 라면 이것을 하나의 괄호에 넣은 새로운 문자열 “(x)”도 VPS 가 된다. 그리고 두 VPS x 와 y를 접합(concatenation)시킨 새로운 문자열 xy도 VPS 가 된다. 예를 들어 “(())()”와 “((()))” 는 VPS 이지만 “(()(”, “(())()))” , 그리고 “(()” 는 모두 VPS 가 아닌 문자열이다.

여러분은 입력으로 주어진 괄호 문자열이 VPS 인지 아닌지를 판단해서 그 결과를 YES 와 NO 로 나타내어야 한다.

입력

입력 데이터는 표준 입력을 사용한다. 입력은 T개의 테스트 데이터로 주어진다. 입력의 첫 번째 줄에는 입력 데이터의 수를 나타내는 정수 T가 주어진다. 각 테스트 데이터의 첫째 줄에는 괄호 문자열이 한 줄에 주어진다. 하나의 괄호 문자열의 길이는 2 이상 50 이하이다.

출력

출력은 표준 출력을 사용한다. 만일 입력 괄호 문자열이 올바른 괄호 문자열(VPS)이면 “YES”, 아니면 “NO”를 한 줄에 하나씩 차례대로 출력해야 한다.

예제입력

6
(())())
(((()())()
(()())((()))
((()()(()))(((())))()
()()()()(()()())()
(()((())()(

예제출력

NO
NO
YES
NO
YES
NO

📖내가 작성한 Code

import sys


class Node:
    def __init__(self,data):
        self.node = data
        self.next = None


class Stack:
    def __init__(self):
        self.top_node = None
        self.balance = 0

    def push(self, data):
        if self.top_node:
            if not self.top_node.node + data:
                self.balance += data
                self.top_node = self.top_node.next
            else:
                node = Node(data)
                node.next = self.top_node
                self.top_node = node
                self.balance += data
        else:
            self.top_node = Node(data)
            self.balance += data


def find_vps(lst):
    new_stack = Stack()
    for ps in lst:
        match ps:
            case '(':
                new_stack.push(1)
            case ')':
                new_stack.push(-1)
        if new_stack.balance < 0:
            return 'NO'
    if new_stack.balance == 0:
        return 'YES'
    else:
        return 'NO'


def main():
    input = sys.stdin.readline
    test_case = int(input())

    for case in range(test_case):
        print(find_vps(input().strip()))


if __name__ == "__main__":
    main()

✍️풀이과정

자료구조로 문제라고 일부러 스택으로 해보려고 했음. 지금 생각해보면 왜 balance를 변화 시켰는지 이해를 할 수 없었음.


🧠 코드 리뷰

  1. 논리 오류 : balance 변경에 오류.

  2. 가독성 개선 : not self.top_node.node + data 보다는 if self.top_node.node + data == 0 명시적으로 개선

  3. lst가 아닌 s와 char로 대체 권장


🛠️AI 개선 코드

    import sys

    class Node:
        def __init__(self, data):
            self.data = data
            self.next = None

    class Stack:
        def __init__(self):
            self.top_node = None

        def push(self, data):
            if self.top_node and (self.top_node.data + data == 0):
                self.top_node = self.top_node.next
            else:
                node = Node(data)
                node.next = self.top_node
                self.top_node = node

        def is_empty(self):
            return self.top_node is None

    def find_vps(s):
        stack = Stack()
        for char in s:  # 문자열을 직접 순회
            if char == '(':
                stack.push(1)
            elif char == ')':
                stack.push(-1)
            # 여는 괄호가 없는데 닫는 괄호가 나왔을 경우
            if stack.is_empty() and char == ')':
                return 'NO'
        return 'YES' if stack.is_empty() else 'NO'

    def main():
        input = sys.stdin.readline
        test_case = int(input())
        for _ in range(test_case):
            s = input().strip()
            print(find_vps(s))

    if __name__ == "__main__":
        main()


💻결과

백준문제 보러가기


🖱️참고 링크

자료구조 Python 공식문서

profile
우물 안에서 무언가 만드는 사람

0개의 댓글