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

소스코드

스택을 활용한 풀이

class Stack:
    def __init__(self):
        self.data = []

    def push(self, item):
        self.data.append(item)

    def pop(self):
        if self.size() == 0:
            return -1
        else:
            return self.data.pop()

    def size(self):
        return len(self.data)

    def isEmpty(self):
        if self.size() == 0:
            print('YES')
        else:
            print('NO')

t = int(input())

for i in range(t):
    stack = Stack()
    vps = input()
    flag = True
    for j in vps:
        if j == '(':
            stack.push(j)
        else:
            pop = stack.pop()
            if pop == -1:
                print('NO')
                flag = False
                break
    if flag:
        stack.isEmpty()

풀이

vps에 문자열을 받아 for 적용. ‘(‘가 나오는 경우 스택에 넣어준다. ‘)’가 나오는 경우에는 스택에 있는 ‘(‘을 pop해 없앤다.

여기서 NO가 나오려면

  1. ’)’의 쌍이 되는 ‘(‘이 없거나
  2. ’(‘의 쌍이 되는 ‘)’이 없어야 한다.

1은 pop을 하려는데 스택에 더이상 원소가 없을 때이다. self.size() == 0이 되어 -1을 return. ‘NO’를 출력, flag를 False로 바꿔 for문을 break한다.

2는 스택에 ‘(‘가 남아있지만 vps 반복문이 끝나는 경우. stack.isEmpty()로 비어있지 않으면 ‘NO’를 출력한다.

vps for문이 끝났는데 스택이 비어있으면 모든 쌍이 맞으므로 ‘YES’를 출력한다.

간단한 풀이

t = int(input())

for i in range(t):
    vps = input()
    cnt = 0
    for j in vps:
        if j == '(':
            cnt += 1
        else:
            cnt -= 1

        if cnt < 0:
            print('NO')
            break
    if cnt == 0:
        print('YES')
    elif cnt > 0:
        print('NO')