[백준] 2504번_괄호의값

문제 링크 : https://www.acmicpc.net/problem/2504

2504번: 괄호의 값

stack = []
bracket = input()

stack.append(bracket[0])
point = 1
while point<len(bracket):
    new_ele = bracket[point]
    if new_ele == ")":
        try:
            if stack[-1] == "(":
                stack.pop()
                stack.append(2)
            elif str(stack[-1]).isdigit() and stack[-2] == "(":
                tmp = stack.pop()
                stack.pop()
                stack.append(tmp*2)
        except IndexError:
            stack.append(new_ele)
    elif new_ele == "]":
        try:
            if stack[-1] == "[":
                stack.pop()
                stack.append(3)
            elif str(stack[-1]).isdigit() and stack[-2] == "[":
                tmp = stack.pop()
                stack.pop()
                stack.append(tmp*3)
        except IndexError:
            stack.append(new_ele)
    else:
        stack.append(new_ele)
    if len(stack)>=2:
        if str(stack[-1]).isdigit() and str(stack[-2]).isdigit():
            tmp = stack.pop()
            tmp += stack.pop()
            stack.append(tmp)
    point += 1

flag = True  
for ele in stack:
    if ele == "(" or ele==")" or ele=="[" or ele=="]":
        print(0)
        flag = False
        break
if flag:
    print(stack[0])
  

하드 코딩이라 좀 더러워서 다시 고쳐봐야지