[python] 백준 - 2504. 괄호의 값

🤔문제 해결

• S2 | 스택, 자료구조, 구현, 재귀

괄호문제는 항상 스택으로 풀었다.

• 먼저 올바른 괄호인지 체크하자! 그 다음
• 스택을 이용해 괄호 짝을 맞추면서 ()일 때 2, []일 때 3을 계산해준다.
• 만약 스택의 마지막에 숫자가 있다면 곱해준다.
• 만약 스택에 숫자가 연속으로 나열되면 더해준다.

 

 

💻소스 코드

def is_right(s):
    stack = []
    for ss in s:
        if ss == '(':
            stack.append(ss)
        elif ss == ')':
            if stack and stack[-1] == '(':
                stack.pop()
            else:
                return False
        elif ss == '[':
            stack.append(ss)
        elif ss == ']':
            if stack and stack[-1] == '[':
                stack.pop()
            else:
                return False
    if stack:
        return False
    return True


def my_stack(s):
    stack = []
    for c in s:
        print(stack)

        if c == '(':
            stack.append(c)
        elif c == '[':
            stack.append(c)
        elif c == ')':
            tmp = 2
            while True:
                t = stack.pop()
                if type(t) == int:
                    tmp *= t
                elif t == '(':
                    stack.append(tmp)
                    break
        elif c == ']':
            tmp = 3
            while True:
                t = stack.pop()
                if type(t) == int:
                    tmp *= t
                elif t == '[':
                    stack.append(tmp)
                    break
        tmp = 0
        while stack:
            t = stack.pop()
            if type(t) == int:
                tmp += t
            else:
                stack.append(t)
                break
        if tmp:
            stack.append(tmp)


    return stack[-1]


S = input()
if is_right(S):
    print(my_stack(S))
else:
    print(0)

 

📕문제 확인

출처: BACKJOON ONLINE JUDGE

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

2504번: 괄호의 값