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

예제 입력 1 복사

6
(())())
(((()())()
(()())((()))
((()()(()))(((())))()
()()()()(()()())()
(()((())()(

예제 출력 1 복사

NO
NO
YES
NO
YES
NO

예제 입력 2 복사

3
((
))
())(()

예제 출력 2 복사

NO
NO
NO

풀이

스택의 LIFO(Last In First Out) 특성을 이용해 풀 수 있는 문제입니다. 입력받은 문자열을 하나씩 스택에 넣으면서 넣을 문자와 가장 최근에 넣은 문자를 비교하고, 두 문자를 합쳤을 때 올바른 괄호가 만들어진다면 스택에서 지워나가면 됩니다. 예제 입력 1의 가장 첫 번째 예시를 그림으로 표현하면,

입력값 (

입력값 (

입력값 )

입력값 )

입력값 ) - 결과

 첫 번째 예시에서 가장 마지막 문자까지 입력받으면 위와 같은 스택이 만들어집니다. 모두 올바른 괄호이면 스택에 아무것도 남아있지 않을 것이고, 올바르지 않은 괄호라면 스택에 남은 괄호가 있겠죠? 따라서 스택이 비어있다면 YES를 출력하고 비어있지 않다면 NO를 출력합니다. 코드와 대략적인 코드 풀이는 아래와 같습니다.

# 괄호

def check_vps(ps):
    temp = []
    for i in ps:
        if i == '(':
            temp.append(i)
        else:
            if len(temp) == 0 or temp[-1] == ')':
                temp.append(i)
            else:
                temp.pop(-1)

    if len(temp) == 0:
        print('YES')
    else:
        print('NO')

n = int(input())

for i in range(n):
    # Input Parenthesis String
    ps = input()
    check_vps(ps)

1. 입력 데이터의 수 n을 입력받습니다.
2. n만큼 반복하며 ps(괄호 문자열)을 입력받습니다.
3. check_vps에 입력받은 ps를 인자로 넘겨줍니다.
   1. 스택으로 사용할 리스트 temp를 초기화해줍니다.
   2. 입력받은 문자열의 값을 하나씩 불러옵니다.
      1. 만약 문자열의 값이 ‘**(’**라면 temp에 그냥 넣어줍니다.
      2. 문자열의 값이 ‘)’일 때, temp에 아무것도 들어있지 않거나 바로 이전 문자열이 ‘)’ 라면 ‘)’를 넣어줍니다.
      3. 그렇지 않다면 스택에서 마지막 요소를 삭제합니다.
   3. 만약 스택 temp가 비어있다면, 모든 괄호의 모양이 올바르게 구성되어 있으므로 ‘YES’를 출력합니다. 그렇지 않다면 ‘NO’를 출력합니다.