[알고리즘 : 자료구조] #9012 괄호

🔗 문제 원본 : https://www.acmicpc.net/problem/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”를 한 줄에 하나씩 차례대로 출력해야 한다.

내가 푼 방식

문자열을 리스트로 치환하고, 리스트의 요소를 stack이라는 또 다른 리스트에 하나씩 담아 이전 요소들과 비교하여 '('와 ')' 쌍이 만날 경우, 해당하는 stack 리스트의 요소를 pop() 해주었다. 이 때, stack의 길이가 최종적으로 0이 되면 'YES'를, 초과하면 'NO'를 출력했다.

import sys

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

for t in range(T):
    command = sys.stdin.readline().strip()
    stack = []
    for c in command:
        if c == '(':
            stack.append(c)
        elif c == ')':
            if len(stack) != 0 and stack[-1] == '(':
                stack.pop()
            else:
                stack.append(c)

    if len(stack) == 0:
        print('yes')
    else :
        print('no')

최대한 스택 구조를 생각하면서 짜긴 했는데 일단 채점에서는 틀리다고 나온 관계로... 나중에 수정할 예정... 예제대로 출력되긴 한다.