백준 / 9012번 괄호 / Python 파이썬

 

*문제 출처는 백준에 있습니다.

문제 제목: 괄호

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

---

나의 풀이

n = int(input())

cnt = 0
def solution(arr):
    stack = []

    if arr[0] == '(': # 첫 괄호가 '('이면 스택에 쌓는다.
        stack.append(arr[0])
    else: # 첫 괄호가 '('가 아니면 올바르지 않은 괄호로 NO를 출력
        answer = "NO" 
        return answer

    for i in range(1,len(arr)):
        if arr[i] == '(': 
            stack.append(arr[i]) #'('는 스택에 쌓는다
        elif arr[i] == ')': #만약 ')' 괄호라면 
            if stack == []: #스택이 비어있으면 쌓는다
                stack.append(arr[i])
                break # for문을 빠져나간다
            if stack[-1] == '(': #만약 스택에 마지막괄호가 '('라면
                stack.pop() # 스택의 마지막 괄호'(' 를 pop해준다. (올바른 괄호이므로)

    if stack == []: #스택이 비어있으면 YES 
        answer = "YES"
    else: # 스택이 비어있지 않으면 NP
        answer = "NO"
    return answer

while cnt != n:
    ps = input()
    print(solution(ps))
    cnt += 1

 

 

---

※ 알아야 할 것

- '(' 이면 스택에 쌓고, ')'를 만나 '()' 가 완성되면 스택에 넣었던 '(' 괄호를 pop해주는 구조를 이용하였다.