[Python] 백준 9012번 - 괄호

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

 

---

스택을 사용하는 유명한 문제다

입력되는것은 단 두 가지 '(' 와 ')' 둘 뿐이니 스택을 사용하여 풀기 쉽다.

 

n = int(input())

arr = []
for _ in range(n):
  arr = input()
  stack = []
  flag = "YES"
  for i in arr:
    if i == '(':
      stack.append(i)
    else:
      if len(stack) == 0:
        flag = "NO"
        break
      else:
        if stack.pop() == i:
          flag = "NO"
          break
  if len(stack) > 0:
    print("NO")
  else:
    print(flag)

---

다른 풀이로는

 

위처럼 굳이 배열에서 직접 뽑아서 스택을 쌓는 게 아닌 더 간단한 방법이 있었다.

 

다른 사람의 코드이다.

 

from sys import stdin

n = int(input())
for _ in range(n):
    str_ = stdin.readline().strip()
    stack = 0
    for chr_ in str_:
        if chr_ == '(':
            stack += 1
        else:
            stack -= 1
            if stack < 0:
                break
    if stack == 0:
        print('YES')
    else:
        print('NO')