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

구현

t = int(input())

for i in range(t):
    vps = str(input())
    stack = [] 
    state = True
    
    for ch in vps:
        if(ch == '('):
            stack.append('(')
        elif(ch == ')'):
            if stack:
                stack.pop()
            else:
                state = False
                break
            
    if stack:
        state = False
    
    print("YES" if state else "NO")

풀이

이 문제는 괄호의 짝이 올바른지 확인하는 문제로 스택을 사용하여 해결할 수 있습니다. ‘(‘는 스택에 넣고, ‘)’가 나오면 스택에서 하나를 제거합니다. 이때 스택이 비어있는 상태에서 ‘)’가 나오거나 순회가 끝난 후에도 스택에 괄호가 남아있다면 올바른 괄호 문자열이 아닙니다. 각 테스트마다 stack과 state를 초기화하고, state 값을 기준으로 “YES” 또는 “NO”를 출력하도록 했습니다.