[백준] 9012 : 괄호 - 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

 

---

풀이

여는 괄호 - (
닫는 괄호 - )
괄호 문자열이 문제의 조건에 맞게 구성되려면 닫는 괄호가 나왔을 경우에 앞에 짝을 이루는 여는 괄호가 있어야 한다.
여는 괄호가 나오면 스택에 push, 닫는 괄호가 나오면 pop을 수행한다.
닫는 괄호가 나왔을 때 스택이 비어있어 pop 연산이 불가능하다면 올바른 괄호 문자열이 아님으로 "NO"를 출력하고 다음 문자열을 순회한다.
문자열을 순회하는 반복문이 정상적으로 종료되었을 때 스택에 원소가 남아있다면 여는 괄호와 짝을 이루는 닫는 괄호가 부족하다는 것이므로 "NO"를 스택이 비었다면 "YES"를 출력한 후에 다음 문자열을 순회한다.
반복문이 정상적으로 종료되었을 때 스택에 원소가 남아있다면 "NO" 그렇지 않다면 "YES"를 출력하는 코드를 어떻게 구현할지 방법을 찾아보며 for-else 문법을 알게 되었다.
for문 뒤에 else블록을 사용하면 for문이 정상적으로 종료되었을 때는 else블록이 실행되고 강제종료(break) 되었다면 실행되지 않는다.

---

코드

N = int(input())#받을 괄호문자열의 수

for i in range(N):
    stack = []#스택으로 사용할 리스트
    brackets = input()#괄호 문자열 받기

    for bracket in brackets:
        if bracket == "(": 
            stack.append(bracket) #"(" 나오면 스택에 추가

        elif bracket == ")":
            if(len(stack) == 0):  #스택이 비어있는데 닫는 괄호가 나온 경우
                print("NO")
                break

            #닫는 괄호가 나왔고 스택이 비어있지 않다면 스택의 여는 괄호 제거
            stack.pop()

    else: 
        """
        for문 뒤에 else블록 사용
        for문이 정상적으로 종료되었을 때 else블록 실행
        break로 강제종료 되었다면 실행 X
        """
        if len(stack) == 0:
            print("YES")
        else:
            print("NO")