[🥈4 / 백준 9012 / 파이썬] 괄호

 

9012번: 괄호

 

문제

괄호 문자열(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”를 한 줄에 하나씩 차례대로 출력해야 한다. 

 

예제 입력 1  6 (())()) (((()())() (()())((())) ((()()(()))(((())))() ()()()()(()()())() (()((())()(	예제 출력 1  NO NO YES NO YES NO
예제 입력 2  3 (( )) ())(()	예제 출력 2  NO NO NO

 

풀이

 

스택을 이용해 VPS를 찾는 문제이다.

 

VPS가 아닌 경우란 두 가지로 정할 수 있다.

 

1. 괄호가 열리기도 전에 닫음

2. 열린 괄호가 닫히지 않음

 

괄호의 특성상 무조건 괄호를 여는 것이 선행되고, 그 이후에 닫힘이 따라와야 한다.

 

문자열의 처음부터 짚어가며 괄호가 열릴 때마다 스택에 push, 닫히면 pop을 하며 짝을 찾아주자.

 

여기서, 스택이 비었는데 pop을 한다거나 문자열이 끝났는데 스택이 비어있지 않다면 VPS가 아닌 것이 된다.

 

import sys
input = sys.stdin.readline
for i in range(int(input())):
    stack = []
    for i in input().rstrip():
        if i == "(":
            stack.append(i)
        else:
            try:
                stack.pop()
            except:
                print("NO")
                break
                # 괄호를 열기도 전에 닫음
    else:
        print("NO" if stack else "YES")
        # 스택이 안 비어있는 경우 NO 출력