[백준/python] 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”를 한 줄에 하나씩 차례대로 출력해야 한다. 

 

정답

import sys
T = int(sys.stdin.readline())

answer = []
for i in range(T):
    vps = list(input())
    stack = []
    a = "YES"
    for n in vps:
        if n == ")":
        	# stack에 "("가 없는 경우
            if not stack:
                a = "NO"
                break
            ps = stack.pop()
		# n == "("
        else:
            stack.append(n)
    # 짝지어질 "("가 없는 경우
    if stack:
        a = "NO"
    answer.append(a)


for i in answer:
    print(i)

---

여는 괄호가 나오면 stack에 넣어주고 닫는 괄호가 나왔을 때 stack에서 pop()해주는 방식이다.

이 과정을 수행할 때 NO가 나올 경우를 걸러 a에 "NO"를 저장한다.

 

"NO"가 나와야 하는 경우

1. "("가 나오기 전에 ")"가 나오는 경우 : ")"가 들어왔을 때 stack이 비어있을 경우
2. ")"와 짝지을 "("가 없는 경우 : ")"가 들어왔을 때 stack이 비어있을 경우
3. "("와 짝지을 ")"가 없는 경우 : 문장의 모든 괄호를 확인한 후에도 stack에 "("가 남아있을 경우

아래 포스팅은 이 문제와 유사한 예제이다.

2021.03.12 - [🔍알고리즘/백준] - [백준/python] 4949번 : 균형잡힌 세상

[백준/python] 4949번 : 균형잡힌 세상