[백준 4949번] 균형잡힌 세상 -python

문제 링크

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

4949번: 균형잡힌 세상

 

풀이 

원래는 defaultdic(int)로 풀려고 했다 

- i가 '(' 이나 '[' 이면 dic[i] += 1 

- i가 ')' 이나 ']' 이면 dic[i] -= 1

이 때, dic[i]의 값이 음수이면 no

- dic['(']나 dic['[']가 0이 아니면 no

- 둘 다 0 이면 yes 

그러나 이렇게 풀면 아래와 같은 경우 yes로 판단.. 

Help( I[m being held prisoner in a fortune cookie factory)].

#틀린 코드
from collections import defaultdict

while True:
  s = input()
  result = True
  dic = defaultdict(int)

  if s == '.':
    break

  for i in s:
    if i == '(' or i =='[':
      dic[i] += 1
    elif i == ']':
      dic['['] -= 1 
      if dic['['] < 0 :
        print('no')
        result= False
        break
    elif i == ')':
      dic['('] -= 1 
      if dic['('] < 0 :
        print('no')
        result = False
        break

  if result :
    if dic['('] != 0 or dic['['] != 0 :
      print('no')
    else:
      print('yes')

따라서 stack으로 변경해서 풀었다. 

stack = []

while True:
  s = input()
  result = True
  stack = []

  if s == '.':
    break

  for i in s:
    # stack에 추가 
    if i == '(' or i =='[':
      stack.append(i)
      
    # stack이 비어있지 않고 쌍이 맞으면 pop 틀리면 no 출력 
    elif i == ']':
      if len(stack) > 0 and stack[-1] == '[':
        stack.pop()
      else :
        print('no')
        result = False
        break
        
    # stack이 비어있지 않고 쌍이 맞으면 pop 틀리면 no 출력 
    elif i == ')':
      if len(stack) > 0 and stack[-1] == '(':
        stack.pop()
      else :
        print('no')
        result = False
        break
  
  if result :
    # 스택에 괄호가 남아있으면 No 
    if stack:
      print('no')
    else:
      print('yes')