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

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

4949번: 균형잡힌 세상

def func(inp):
    data = []
    for i in inp:
        if i == "(" or i == "[":
            data.append(i)
        elif i == "]":
            if data == []:
                return "no"
            elif data.pop() != "[":
                return "no"
        elif i == ")":
            if data == []:
                return "no"
            elif data.pop() != "(":
                return "no"
    if data == []:
        return "yes"
    else:
        return "no"
    
while True:
    inp = input()
    if inp == '.':
        break
    print(func(inp))