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

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

4949번: 균형잡힌 세상

 

1. 내가 작성한 코드

while True:
  arr = list(input())

  if arr == ["."]: #.만나면 종료
    break

  stack = []
  
  for i in arr:
    if i == "[" or i == "(":
      stack.append(i)

    elif i == "]":
      if len(stack) != 0 and stack[-1] == "[":
        stack.pop()
      
      else:
        stack.append("]")
        break

    elif i == ")":
      if len(stack) != 0 and stack[-1] == "(":
        stack.pop()
      
      else:
        stack.append(")")
        break
    
  
  if len(stack) > 0:
    print("no")
  else:
    print("yes")