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

문제 출처

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

---

문제 풀이

열린 괄호( 와 닫힌 괄호 ) 의 짝이 맞아야 올바른 문자열이라고 판정할 수 있다.

 

스택으로 짝의 대응을 확인할 수 있다.

( 괄호를 만나면 스택에 삽입하고, ) 괄호를 만나면 삽입했던 ( 괄호를 스택에서 빼낸다.

 

주어진 문자열을 끝까지 연산을 수행했을 때 스택에 무언가 남아있다면 올바르지 못한 문자열이 될 것이고, 스택에 아무것도 남아있지 않다면 올바른 문자열이 될 것이다. 또한 스택에 남아있는 ( 괄호가 없음에도 ) 괄호를 만나는 경우에도 올바르지 못한 문자열로 판정한다.

 

 

스택은 FILO 이므로 배열의 맨 끝 원소만 가지고 삽입 삭제 연산을 수행하는 자료구조다.

즉, 큐를 구현할때에는 deque로 구현하는게 유리했다면, 스택은 리스트로 구현해도 괜찮다. 끝 값만 삽입 삭제 연산을 수행하면 deque나 리스트나 동일하게 O(1) 시간복잡도를 가지기 때문이다.

 

작성 코드

t = int(input())
for _ in range(t):
    stack = []
    vps = ''
    parenthesis = input()
    for p in parenthesis:
        if p == '(':
            stack.append(p)
        elif p == ')':
            if len(stack) > 0:
                stack.pop()
            else:
                vps += 'NO'
                break
    
    if vps == '' and len(stack) == 0:
        vps += 'YES'
    elif vps == '' and len(stack) > 0:
        vps += 'NO'
    
    print(vps)

실행시간: 80ms