[백준] 9012번: 괄호 (Python/파이썬)

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

 

풀이과정

0. 접근

VPS 여부를 판단하는 문제로, 여는 괄호와 닫는 괄호를 짝지어 소거하였을 때, 남는 괄호가 없다면 VPS라고 판단할 수 있다.

이에 여는 괄호는 stack에 담고, 닫는 괄호는 stack의 여는 괄호와 짝을 지어 소거하는 방식을 사용하기로 했다. 

이 때 여는 괄호를 보관하는 자료구조로 stack을 사용하는 이유는,
여는 괄호와 닫는 괄호를 짝짓는 과정에서 닫는 괄호 중 첫 번째 것은 여는 괄호 중 마지막 것과 짝을 지어야 하기 때문에 LIFO principle이 적용된 stack을 활용한 것이다.

 

1. 런타임 에러 (IndexError)

T = int(input())

for i in range(T):

    str = input()
    str_list = list(str)
    stack = []
    
    for j in str_list:
        if j == "(":
            stack.append(j)
        else:
            stack.pop(0)
    
    if len(stack) == 0:
        print("YES")
    else:
        print("NO")

stack.pop(0)에서 IndexError가 발생한 것 같다. stack이 비어있는 경우에는 첫 번째 원소 (index 0) 가 없기 때문이다.

 

2. 오답

T = int(input())

for i in range(T):

    str = input()
    str_list = list(str)
    stack = []
    
    for j in str_list:
        if j == "(":
            stack.append(j)
        elif len(stack) == 0:
            print("NO")
            break
        else:
            stack.pop()
 
    if j == str_list[-1] and len(stack) == 0:
        print("YES")
    else:
        print("NO")

닫는 괄호의 케이스에서 stack이 비어있는 경우를 추가로 고려했다. 

하지만 안쪽 for문에서 마지막 j가 elif문을 마주친 경우, for문을 빠져나와서 한 번 더 YES를 출력하는 문제가 있어 오답 처리 되었다.

( ex.(())(() )

 

3. 정답

T = int(input())

for i in range(T):

    str = input()
    str_list = list(str)
    stack = []
    earlychk = False
    for i, j in enumerate(str_list):
        if j == "(":
            stack.append(j)
        elif j == ")":
            if len(stack) == 0:
                print("NO")
                earlychk = True
                break
            else:
                stack.pop()
 
    if len(stack) == 0 and earlychk == False:
        print("YES")
    elif earlychk == False:
        print("NO")

안쪽 for문에서 print를 한 경우에는 바깥쪽 for문에서 다시 print를 하지 않도록 하기 위해서, earlychk라는 변수를 도입했다. 

만약 이 문제가 VPS 여부를 확인하는 '함수'를 만드는 문제라고 가정했을 때 return과 같은 역할을 할 수 있는 도구를 고민하다가 생각해낸 방법이다.