[python] BOJ 9012번 : 괄호

난이도 : 실버 IV

문제

괄호 문자열(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

input=sys.stdin.readline

T=int(input())

for _ in range(T):
    words=input().strip()
    stacks=[]
    for word in words:
        if word=='(':
            stacks.append('(')
        elif word==')':
            if len(stacks)>=1 and stacks[-1]=='(':
                stacks.pop(-1)
            else:
                stacks.append(')')
    if len(stacks)==0:
        print("YES")
    else:
        print("NO")

 

 

 

문제 링크

https://www.acmicpc.net/problem/9012

9012번: 괄호