백준: 2504 괄호의 값

문제

2504번: 괄호의 값

 

문제 풀이

 올바른 괄호인지 판단하면서, `()`인 경우에는 2로 치환하고 `[]`인 경우에는 3으로 치환하여 계산하는 과정이 필요하다. 또한 치환된 값이 `()` 또는 `[]` 사이에 있다면 2나 3이 곱해지게 된다. 처음에는 해당 과정을 하나로 처리하려고 했는데, 코드 읽기가 힘들어 따로 판단하는 것이 좋다고 생각했다.

 

• 입력으로 주어진 괄호들이 올바른 괄호인지 판단한다.
  • 올바른 괄호는 프로그래머스: 올바른 괄호와 같이 작성하면 판단할 수 있다.
• 각 괄호들을 계산하는 방법은 다음과 같다.
  • stack에 `()`와 `[]`가 만날 경우 변환되는 숫자를 반영한다.
  • 숫자와 괄호가 만나는 상황을 고려하여, sub total을 통해 값을 반영하여 stack에 넣는다.
  • 모든 입력 값에 대한 처리가 끝난 후, 합을 계산한다.
• 딕셔너리를 활용하여, 가독성을 높이고자 했는데 높아진지는 모르겠다.

 

코드

from sys import stdin

case = {
    '(': ')',
    '[': ']'
}


def possible() -> bool:
    stack = []

    for bracket in string:
        if bracket in case.keys():
            stack.append(bracket)
        elif bracket in case.values():
            if stack and bracket == case[stack[-1]]:
                stack.pop()
            else:
                return False

    return True if not stack else False


def change_bracket(value) -> None:
    cur_stack, cur_bracket = value
    sub_total = 0
    num = 2 if cur_bracket == ')' else 3

    while cur_stack:
        top = cur_stack.pop()
        if type(top) == str and cur_bracket == case[top]:
            cur_num = num if not sub_total else num * sub_total
            cur_stack.append(cur_num)
            break

        sub_total += top


def calculate() -> int:
    stack = []

    for bracket in string:
        if bracket in case.keys():
            stack.append(bracket)
        elif stack and bracket in case.values():
            change_bracket([stack, bracket])

    return sum(stack)


if __name__ == '__main__':
    string = stdin.readline().strip()
    print(calculate() if possible() else 0)