[Algorithm] 백준 9012

문제

괄호 문자열(Parenthesis String, PS)은 두 개의 괄호 기호인 ‘(’ 와 ‘)’ 만으로 구성되어 있는 문자열이다. 그 중에서 괄호의 모양이 바르게 구성된 문자열을 올바른 괄호 문자열(Valid PS, VPS)이라고 부른다. 한 쌍의 괄호 기호로 된 “( )” 문자열은 기본 VPS 이라고 부른다. 만일 x 가 VPS 라면 이것을 하나의 괄호에 넣은 새로운 문자열 “(x)”도 VPS 가 된다. 그리고 두 VPS x 와 y를 접합(concatenation)시킨 새로운 문자열 xy도 VPS 가 된다. 예를 들어 “(())()”와 “((()))” 는 VPS 이지만 “(()(”, “(())()))” , 그리고 “(()” 는 모두 VPS 가 아닌 문자열이다.

여러분은 입력으로 주어진 괄호 문자열이 VPS 인지 아닌지를 판단해서 그 결과를 YES 와 NO 로 나타내어야 한다.

풀이

내가 생각한 알고리즘

생각보다 꽤나 까다로운 문제였다.

주어진 알고리즘은 주어진 문자열이 올바른 괄호열인지 여부를 판별하는 것입니다. 여기서 "올바른 괄호열"은 '('와 ')'의 괄호가 올바르게 짝을 이루고 있는지를 의미합니다.

알고리즘은 다음과 같이 작동합니다:

1. 입력으로 주어진 괄호열을 한 줄씩 읽어들입니다.
2. 주어진 문자열을 한 글자씩 순회하면서, '('를 만나면 obj에 추가하고, ')'를 만나면 obj에서 '('를 꺼냅니다.
3. 이 과정을 모든 문자열에 대해 수행한 후에 이 obj비어있으면 올바른 괄호열입니다.
4. obj이 비어있지 않다면 올바른 괄호열이 아닙니다.

간단히 정리하자면, 이 알고리즘은 입력으로 주어진 문자열이 올바른 괄호열인지를 판별하는 것으로, '('를 만나면 스택에 추가하고 ')'를 만나면 스택에서 '('를 제거하여 짝을 이루는지 확인합니다. 모든 문자열에 대해 이를 반복하고, 스택이 비어있으면 올바른 괄호열로 판별합니다.

코드 구현

import sys

N = int(sys.stdin.readline().rstrip())

def is_vap(string):
    obj = []
    for j in range(0, len(string)):
        if len(obj) == 0 :
            obj.append(string[j])
        else:
            if obj[-1] == '(' and string[j] == ')':
                obj.pop()
            else:
                obj.append(string[j])

    if len(obj) == 0:
        return True
    return False

for i in range(N):
    str = sys.stdin.readline().rstrip()
    if is_vap(str):
        print('YES')
    else:
        print('NO')

Lesson Learn

이거는 어떻게 하면 주어진 문자열에서 판별할 수 있을지 고민해보는 것이 굉장히 중요했던 문제이다.
정답이 될 수 있는 규칙을 찾고, 알고리즘을 짜는 연습을 하자