[백준-2504] 괄호의 값 / Python

2504번: 괄호의 값

import sys

input = sys.stdin.readline

s = input().rstrip()
stack = []
answer = 0

for i in s:
    if i == ")":
        t = 0
        while len(stack) != 0:
            top = stack.pop()
            if top == "(":
                if t == 0:
                    stack.append(2)
                else:
                    stack.append(2 * t)
                break
            elif top == "[":
                print(0)
                exit(0)
            else:
                t = t + int(top)
    elif i == "]":
        t = 0
        while len(stack) != 0:
            top = stack.pop()
            if top == "[":
                if t == 0:
                    stack.append(3)
                else:
                    stack.append(3 * t)
                break
            elif top == "(":
                print(0)
                exit(0)
            else:
                t = t + int(top)
    else:
        stack.append(i)

for i in stack:
    if i == "(" or i == "[":
        print(0)
        exit(0)
    else:
        answer += i

print(answer)

 

해설

풀다가 해결을 못해서 결국 구글링을 했다.. 정확한 해설과 답안 코드는 IT-garden 블로그를 보고 참조하는게 낫다..