[Python] 백준 #4949. 균형잡힌 세상

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

4949번: 균형잡힌 세상

 

# 4949 - Python 3

import sys
from collections import deque

input = sys.stdin.readline

while True:
    x = input().rstrip()
    s = deque()                             # '(', '['를 저장하는 deque
    cnt = 0

    if x[0] == ".":
        break
    
    for i in range(0, len(x)):
        if x[i] == '(':
            s.append(x[i])
        elif x[i] == '[':
            s.append(x[i])
        elif x[i] == ')':
            if not s or s[-1] == '[':       # 가장 최근에 s에 input된 것이 '['이면 안 됨
                cnt += 1
            else:
                s.pop()
        elif x[i] == ']':
            if not s or s[-1] == '(':       # 가장 최근에 s에 input된 것이 '('이면 안 됨
                cnt += 1
            else:
                s.pop()

    if not s and cnt == 0:
        print("yes")
    else:
        print("no")