[파이썬] 백준 - 괄호 - 9012

문제

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

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

입력

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

출력

NO
NO
YES
NO
YES
NO

작성한 코드

import sys

def isValid(s: str) -> str:

    if len(s) % 2 != 0:
        return 'NO'

    default = {
        ')':'('
    }

    open = '('
    stack = []
    for i in s:
        if i in open:
            stack.append(i)
        else:
            if not stack:
                return 'NO'
            if default.get(i) != stack.pop():
                return 'NO'

    if not stack:
        return 'YES'
    else:
        return 'NO'

n = int(sys.stdin.readline())
# strip \n 지우기
lst=[sys.stdin.readline().strip() for i in range(n)]

answer = ['']*len(lst)
for i, v in enumerate(lst):
    answer[i] = isValid(v)

print('\n'.join(answer))

 

백준에서 확인

https://www.acmicpc.net/problem/9012

9012번: 괄호

회고

풀었던 유효한 괄호를 함수화 시켜서 풀었다.

처음에 입력값 어떻게 주는지 몰라서 해맸다.

n = int(sys.stdin.readline())
lst=[sys.stdin.readline().strip() for i in range(n)]