[Algorithm] 백준 9012 괄호 with Python

문제

괄호 문자열(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

---

내가 짠 코드

import sys
ip = sys.stdin.readline
t = int(ip())
for _ in range(t):
    stack = []  #스택 자료구조 사용
    a =ip().rstrip()  #입력받기 
    for i in a:
        if i == '(':  #'('괄호는 무조건 스택에 넣기
            stack.append(i)
        else:
            if stack and stack[-1] =='(':  #')'괄호가 나올 때
                stack.pop()  #짝이 맞으면 스택에서 제거
            else:
                print('NO')  #짝이 안맞으면 NO 출력 후 break
                break
    else:
        print('NO') if stack else print('YES') 
        #스택 완성 후 남아있는 괄호가 있으면 NO, 모두 제거됐으면 YES 출력

스택 자료 구조를 공부할 때 가장 대표적으로 등장하는 괄호 맞추기 문제이다. 

조건이 그리 복잡하진 않아서 하나하나 따져가니 구현이 가능했다. 예전에 강의에서 'for - else' 구문을 사용한 코드를 본 적 있는데 이 문제에서 딱 생각나서 유용하게 사용했다. 'for - else'는 for문을 다 돌고 끝났 후에도 살아남은 item들에 특정 조건을 적용할 때 쓰인다.

stack에 괄호를 담는 과정에서 조건이 맞지 않으면 일차적으로 NO를 출력하고 break를 했다. break를 당하지 않고 살아 남고 stack이 완성됐다면, 거기서 다시 조건을 걸어 YES or NO를 출력했다. 이 방법을 쓰니 나름 깔끔하게 짤 수 있었던 것 같다. 역시 좋은 코드를 많이 보는게 실력 향상의 지름길이구만  : ) 

 

 

 

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

9012번: 괄호