[백준] 4949번 균형잡힌 세상 (파이썬)

4949번: 균형잡힌 세상

문제

나의 생각

• 스택을 구현한다.
• while 반복문 사용하고 . 이 들어오면 break.
• 문장에 "([" 이 들어있으면 stack에 넣는다.
• 문장에 ")]" 이 들어있으면 stack에서 가장 최상위에 있는 것을 꺼낸다. pop()
• ")]"이 나왔는데 stack이 비어있거나 pop() + ")]" 이 "()" , "[]"처럼 쌍을 이루지 않으면 false

나의 풀이

import sys

while True:
    stack = []
    input_data = sys.stdin.readline().rstrip()
    flag = True

    if input_data == ".":
        break
    for data in input_data:
        if data in "([":
            stack.append(data)
        elif data in ")]":
            if not stack or stack.pop() + data not in ["()", "[]"]:
                flag = False

    print("yes") if flag and not stack else print("no")

참고

1. if not stack: stack이 비어있으면 True
2. in 연산자 : 안에 포함되어있으면 True