[파이썬] 단계별 백준 문제 풀기 - 9012번 스택 (괄호)

문제링크

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

 

---

<<소스코드>>

import sys
n = int(sys.stdin.readline()) #VPS를 판별해야하는 갯수

for v in range(n): #for문으로 입력된 괄호문자들을 돌려준다
    words = sys.stdin.readline().rstrip() #words = 입력된 괄호문자들
    stack = [] #빈스택을 만들어 놓고 시작
    
    #괄호문자 하나의 열을 for문으로 돈다.
    #만약 열린괄호라면 스택에 추가한다.
    #닫힌 괄호라면, 스택에 열린괄호 짝꿍이 있는지 확인 후 있다면 pop 탈출!!
    #닫힌 괄호이지만, 스택에 아무것도 없다면 짝이 맞는게 없으니 스택에 혼자 들어간다.
    #스택에 괄호가 하나라도 남아있으면 불완전하다는 뜻이므로 NO 프린트해준다.
    
    for w in words:
        if w == '(':
            stack.append(w)
        elif w == ')':
            if stack and stack[-1] =='(':
                stack.pop()
            else:
                stack.append(w)

    if len(stack) == 0 :
        print("YES")
    else:
        print("NO")

<<결과값>>

6
(())())
NO
(((()())()
NO
(()())((()))
YES
((()()(()))(((())))()
NO
()()()()(()()())()
YES
(()((())()(
NO

---

열린 괄호가 들어오면 스택에 넣어주고, 닫힌 괄호 차례에서, 스택에 마지막으로 들어간 열린 괄호가 있다면 열린괄호를 pop으로 탈출시켜준다. 그렇게 짝이 맞는 괄호들을 스택에서 제거해주면, 스택에 아무것도 안남는다.

그렇다면 문자열은 VPS가 된다. 만약 짝이 맞지 않은 괄호가 스택에 1개라도 남아있다면 VPS가 되지 않으므로 NO를 출력해주었다.

 

 

[실수했던 부분]

첫번째 반례에서 (())()) 여기에서 자꾸 IndexError: list index out of range가 났다. 왜...왜

마지막 " ) " 닫는 괄호가 들어왔을 때, 스택은 비어있었다. 

but if 문에서 그렇다면 근데 스택의 [-1]이 "(" 냐고 조건을 주었으니... stack이 비어있지 않다는 조건을 추가했더니 에러가 나지 않았다. 

그리고 "(" 남은 괄호는 stack에 넣어주어 stack 의 길이가 0이 아니라면 NO 를 프린트하게 했다. 

elif w == ')':
    if stack and stack[-1] =='(':
         stack.pop()
    else:
         stack.append(w)