[ 파이썬(python) ] 백준 2504 - 괄호의 값

📍 백준 2504 - 괄호의 값

문제: 백준 2504 - 괄호의 값

---

💡 나의 풀이

stack에 값을 넣고 순서를 잘 알았더라면 금방 풀 수 있을 텐데, 나한테는 어려웠던 문제였다. 전체적인 흐름은 다음과 같다.

1. 괄호열이 올바른지 올바르지 않은지 검사한다.(def is_checked)
2. 괄호열이 올바르면 def calc_value 함수로 이동하고 괄호열이 올바르지 않다면 print(0)을 출력한다.

이어서 세부적인 흐름을 살펴보자. is_check()는 단순히 올바른지 아닌지만 확인하면 되기때문에 큰 어려움은 없었으나, calc_value() 함수에서 상당히 애를 먹었다. 어떻게 해결했는지 차근차근 살펴보자.

1. 여는 괄호((, [)면 stack에 집어넣는다.
2. 닫는 괄호가 나올때 )의 경우와 ]의 경우를 나눈다.
3. )일때 stack[-1]이 (면 ()가 되므로 괄호열의 값은 2이다. 여기서 중요한점은 stack[-1]값을 2로 바꿔주자. 만약, (가 아니면 해당 값은 숫자기 때문에(숫자인 이유는 올바른 괄호열이라는 전제조건이 붙어있기 때문인데 만약, [가 올 수도 있지않나? 할 수 있지만 그렇게 되면 올바른 괄호열이아니고 이미 def is_checked에서 걸러졌을 것이다.)
4. 5번 규칙에 의해 숫자를 계속 더해준다.(여기서 언제까지 더해주는지가 중요한데, 여는 괄호()가 나올때까지 더해주면 된다.) 계속 더해주다가 여는 괄호(()를 만나게 되면 3번 규칙(‘(X)’ 의 괄호값은 2×값(X) 으로 계산된다.)에 의해 현재까지 누적된 점수에 2를 곱해주면 된다.
5. ]일때도 3번과정과 동일하게 수행하면 된다.(괄호와 점수만 변경해주면 된다.)

 

728x90

s = input()

def is_check(s):    # 올바른 괄호열인지 확인하는 함수
    stack = []
    flag = True

    for i in range(len(s)):
        if s[i] == '(' or s[i] == '[':
            stack.append(s[i])

        else:    # ) ]
            if s[i] == ')':
                if stack and stack[-1] == '(':
                    stack.pop()
                else:
                    flag = False

            else:    # ]
                if stack and stack[-1] == '[':
                    stack.pop()
                else:
                    flag = False

    if not stack and flag:
        return True
    return False

def calc_value(s):    # 괄호의 값을 계산하는 함수
    stack = []
    for i in range(len(s)):
        if s[i] == '(' or s[i] == '[':
            stack.append(s[i])

        else:    # ) ]
            if s[i] == ')':
                if stack[-1] == '(':
                    stack[-1] = 2
                else:    # 올바른 괄호열이기 때문에 숫자만 있다.
                    temp = 0
                    for idx in range(len(stack)-1, -1, -1):    # 괄호 만날 때까지 계속 더해주기 (XY) = X + Y
                        if stack[idx] == '(':    
                            stack[-1] = temp * 2
                            break
                        else:    # ==> type(stack[idx]) == int
                            temp += stack[-1]
                            stack.pop()

            else:    # ]
                if stack[-1] == '[':
                    stack[-1] = 3
                else:    # 올바른 괄호열이기 때문에 숫자만 있다.
                    temp = 0
                    for idx in range(len(stack)-1, -1, -1):    # 괄호 만날 때까지 계속 더해주기 (XY) = X + Y
                        if stack[idx] == '[':     
                            stack[-1] = temp * 3
                            break
                        else:    # ==> type(stack[idx]) == int
                            temp += stack[-1]
                            stack.pop()
    return sum(stack)

if is_check(s):
    print(calc_value(s))
else:
    print(0)