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

---

해결법

문자열로 받은 괄호들에서 "( )" 쌍을 찾아보고, 있으면 "( )"를 제외하는 인덱싱으로 문제를 해결했다.

하지만 stack으로 푸는 기본적인 문제라 해서 두 가지를 작성해보려고 한다.

 

#1. 문자열 인덱싱

import sys; input = sys.stdin.readline

n = int(input())

for _ in range(n):
    ps = input().strip()

    while "()" in ps:
        i = ps.index("()")
        ps=ps[:i]+ps[i+2:] #괄호 인덱스로 삭제하는 과정
    
    # 모두 괄호의 쌍이면 길이가 0
    if len(ps)==0:
        print('YES')
    else:
        print("NO")

#2. STACK

T = int(input())

for i in range(T):
    stack = []
    ps=input()
    for c in ps:
        if c == '(':
            stack.append(c)
        elif c == ')':
            if stack:
                stack.pop()
            else: # stack에 괄호가 없을 경우
                print("NO")
                break
    else: # break문으로 끊기지 않았을 경우
        if not stack: # break문으로 안 끊기고 스택이 비어있다면 모든 짝이 맞음.
            print("YES")
        else: # "(" 개수가 ")"보다 많은 경우 NO이다.
            print("NO")

 

배운점

간단한 문제지만 스택으로 푸는 경우도 있다는걸 생각해야겠다.

sys.readline은 입력값에 개행을 받아서 strip()으로 처리해야하는 걸 잊지 말자!

input은 개행 입력 안받음.