백준 4949번 균형잡힌 세상

여는것과 닫는것의 스택 순서가 안맞아서 실패

---

import collections
import heapq
import sys

check = {
    ")": "(",
    "]": "["
}

while True:
    str_input = sys.stdin.readline().strip()
    if str_input == ".":
        break

    # 괄호들만 스택에 저장하고
    # 여는 스택, 닫는 스택 이렇게 저장해야지
    # 일단 개수가 안맞으면 실패
    # 닫는 스택에서 여는 스택 안맞으면 실패

    open_stack = []
    close_stack = []
    result = "yes"
    for i, s in enumerate(str_input):
        # 여는것만
        if s == "(" or s == "[":
            heapq.heappush(open_stack, (i, s))
        # 닫는것만
        elif s == ")" or s == "]" in s:
            heapq.heappush(close_stack, (i, s))

    # 개수가 안맞아서 실패
    if len(open_stack) != len(close_stack):
        result = "no"

    while close_stack:
        str_close = heapq.heappop(close_stack)
        str_open = heapq.heappop(open_stack)

        # 닫는게 여는거랑 안맞을 시
        if check[str_close[1]] != str_open[1]:
            result = "no"

    print(result)

 

 

해답을 확인 했습니다....

 

https://velog.io/@pmk4236/%EB%B0%B1%EC%A4%80-4949%EB%B2%88-%EA%B7%A0%ED%98%95%EC%9E%A1%ED%9E%8C-%EC%84%B8%EC%83%81-Python

백준 4949번 균형잡힌 세상 - Python