스택 : 괄호의 값(백준 2504번)

▶ 문제 : https://www.acmicpc.net/problem/2504

2504번: 괄호의 값

##### 문제 #####
# 주어진 괄호열을 읽고 그 괄호값을 앞에서 정의한대로 계산하여 출력하는 것이다.
# ‘()’ 인 괄호열의 값은 2이다.
# ‘[]’ 인 괄호열의 값은 3이다.
# ‘(X)’ 의 괄호값은 2×값(X) 으로 계산된다.
# ‘[X]’ 의 괄호값은 3×값(X) 으로 계산된다.
# 올바른 괄호열 X와 Y가 결합된 XY의 괄호값은 값(XY)= 값(X)+값(Y) 로 계산된다.

##### 입력 #####
# 첫째 줄에 괄호열을 나타내는 문자열(스트링)이 주어진다. 단 그 길이는 1 이상, 30 이하이다.

##### 출력 #####
# 첫째 줄에 그 괄호열의 값을 나타내는 정수를 출력한다. 만일 입력이 올바르지 못한 괄호열이면 반드시 0을 출력해야 한다. 

▶ 접근 방법

 

▶ 풀이 코드

import sys

parenthesis = sys.stdin.readline().rstrip()

stack_A = []
stack_B = []
value = 1
paried = True
result = 0


for i in range(len(parenthesis)):
    if parenthesis[i] == '(':
        stack_A.append(i)
        value *= 2
    elif parenthesis[i] == '[':
        stack_B.append(i)
        value *= 3
    elif parenthesis[i] == ')':
        if stack_A:
            if parenthesis[i-1] == '(':
                result += value
            stack_A.pop()
            value //= 2
        else :
            paried = False
            break
    elif parenthesis[i] == ']':
        if stack_B:
            if parenthesis[i-1] == '[':
                result += value
            stack_B.pop()
            value //= 3
        else :
            paried = False
            break

if not stack_A and not stack_B and paried:
    print(result)
else :
    print(0)

 

▶ 관련 링크

2021.11.17 - [알고리즘] - 스택(Stack)이란?

2021.11.17 - [알고리즘] - 스택 : 스택(백준 10828번)

2021.11.17 - [알고리즘] - 스택 : 제로(백준 10773번)

2021.11.17 - [알고리즘] - 스택 : 괄호(백준 9012번)

2021.11.17 - [알고리즘] - 스택 : 막대기(백준 17608번)

2021.11.17 - [알고리즘] - 스택 : 탑(백준 2493번)

2021.11.17 - [알고리즘] - 스택 : 원 영역(백준 10000번)

2021.11.17 - [알고리즘] - 스택 : 크게 만들기(2812번)