백준 2504 [괄호의 값] 파이썬

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

 

아주 아주 아주 해맸다.

# 백준 2504 괄호의 값

lst=list(map(str,input().rstrip())) # 입력값 처리
valid_stack=[] # 유효성 여부 확인 위한 스택
success_flg=True # 괄호의 정상 여부 판단 플래그
multiple_flg=False # 계산 값을 더하기 위한 플래그
temp=1
ans=0
for curr_c in lst:
    if curr_c == '(':
        temp*=2 
        valid_stack.append(curr_c) 
        multiple_flg=True
    elif curr_c =='[':
        temp*=3
        valid_stack.append(curr_c)
        multiple_flg=True
    elif curr_c==')':
        if multiple_flg==True:
            ans+=temp
            multiple_flg=False
        temp//=2
        if valid_stack:
            pair=valid_stack.pop()
            if pair!='(':
                success_flg=False
                break
        else:
            success_flg=False
            break
    elif curr_c==']':
        if multiple_flg==True:
            ans+=temp
            multiple_flg=False
        temp//=3
        if valid_stack:
            pair=valid_stack.pop()
            if pair!='[':
                success_flg=False
                break
        else:
            success_flg=False
            break

if success_flg and not valid_stack:
    print(ans)
else:
    print(0)

그냥 그리디하게 스택으로 해결하면 된다.

간단하지만 간단한 아이디어를 생각해내는게 쉽지않았다.