본문 바로가기
Algorithm/Python

[백준9012번] 괄호

by _sweep 2022. 1. 28.

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

 

 

📝 풀이

import sys
num = int(sys.stdin.readline())
arr = [sys.stdin.readline().strip() for i in range(num)]

stack = []
result = []
isVPS = True

for i in arr:
  for j in i:
    if j == "(":
      stack.append(j)
    else:
      if len(stack) == 0:
        isVPS = False
        result.append("NO")
        break
      stack.pop()
      
  if len(stack) != 0:
    result.append("NO")
    stack = []
    isVPS = True
  elif isVPS and len(stack) == 0:
    result.append("YES")
  else:
    isVPS = True

print(*result, sep='\n')

 

유명한 문제인 올바른 괄호 찾기 문제이다.

(()) 같이 올바르게 짝지어진 괄호라면 YES를, (() 같이 짝이 맞지 않는 괄호라면 NO를 출력한다.

이 문제를 풀 때는 스택을 이용하는 것이 제일 편하다.

 

문제의 풀이과정은 다음과 같다.

  1. 입력받은 괄호 문자열들을 arr에 저장한다.
  2. arr의 문자열들을 순회하며 이중 for문을 이용해 각 문자를 살펴본다.
  3. 해당 문자가 여는 괄호 ( 라면 스택에 넣는다.
  4. 해당 문자가 닫는 괄호 ) 라면 스택에서 마지막 문자를 뺀다.
    • 만약 else문에 진입했을 때 이미 빈 스택이라면 짝이 맞지 않는 것이므로 result에 NO를 더해준다.
    • 이때 이후에 스택의 길이를 판별할 때를 대비해 isVPS에 올바른 괄호가 아니라는 뜻으로 False를 넣어준다.
  5. 괄호 문자열 하나의 순회가 끝날 때마다 스택에 남아있는 것이 있는지 판별한다.
    • 이때 만약 스택에 괄호가 남아있다면 짝이 맞지 않는 것이므로 result에 NO를 더해준 후 스택을 비워준다.
    • 위에서 이미 result에 NO를 추가해 isVPS이 False 상태라면 단순히 isVPS를 True로 바꿔준다.
    • isVPS가 True이고 스택의 길이도 0이라면 올바른 괄호이므로 result에 YES를 넣어준다.
  6. result의 요소들을 한줄씩 출력한다.

 

 

 

 

 

 

 

'Algorithm > Python' 카테고리의 다른 글

[백준1120번] 문자열  (0) 2022.01.31
[백준1764번] 듣보잡  (0) 2022.01.31
[백준1541번] 잃어버린 괄호  (0) 2022.01.28
[백준11656번] 접미사 배열  (0) 2022.01.27
[LeetCode] Group Anagrams  (0) 2022.01.25

댓글