[백준 | 실버2] 괄호의 값(올바른 괄호, 스택, 숫자값 계산)

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

 

문제

2504번: 괄호의 값

4개의 기호 ‘(’, ‘)’, ‘[’, ‘]’를 이용해서 만들어지는 괄호열 중에서 올바른 괄호열이란 다음과 같이 정의된다.

1. 한 쌍의 괄호로만 이루어진 ‘()’와 ‘[]’는 올바른 괄호열이다. 
2. 만일 X가 올바른 괄호열이면 ‘(X)’이나 ‘[X]’도 모두 올바른 괄호열이 된다. 
3. X와 Y 모두 올바른 괄호열이라면 이들을 결합한 XY도 올바른 괄호열이 된다.

예를 들어 ‘(()[[]])’나 ‘(())[][]’ 는 올바른 괄호열이지만 ‘([)]’ 나 ‘(()()[]’ 은 모두 올바른 괄호열이 아니다. 우리는 어떤 올바른 괄호열 X에 대하여 그 괄호열의 값(괄호값)을 아래와 같이 정의하고 값(X)로 표시한다. 

1. ‘()’ 인 괄호열의 값은 2이다.
2. ‘[]’ 인 괄호열의 값은 3이다.
3. ‘(X)’ 의 괄호값은 2×값(X) 으로 계산된다.
4. ‘[X]’ 의 괄호값은 3×값(X) 으로 계산된다.
5. 올바른 괄호열 X와 Y가 결합된 XY의 괄호값은 값(XY)= 값(X)+값(Y) 로 계산된다.

예를 들어 ‘(()[[]])([])’ 의 괄호값을 구해보자. ‘()[[]]’ 의 괄호값이 2 + 3×3=11 이므로 ‘(()[[]])’의 괄호값은 2×11=22 이다. 그리고 ‘([])’의 값은 2×3=6 이므로 전체 괄호열의 값은 22 + 6 = 28 이다.

여러분이 풀어야 할 문제는 주어진 괄호열을 읽고 그 괄호값을 앞에서 정의한대로 계산하여 출력하는 것이다. 

입력

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

출력

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

예제 입력 1 복사

(()[[]])([])

예제 출력 1 복사

28

예제 입력 2 복사

[][]((])

예제 출력 2 복사

0

 

 

문제풀이

(, ), [, ] 가 올바른 괄호
() 값 : 2
[] 값 : 3
() * x = 2 * x
[] * x = 2 * x
숫자숫자 = 숫자+숫자

 

문제해결

1. 여는 괄호는 스택 push
2. 닫는 괄호 나오는 경우 스택 꼭대기가 여는 괄호면 2, 3 넣기
3. 닫는 괄호 나오는 경우 스택 꼭대기가 숫자인 경우, 스택 두번째 꼭대기가 여는 괄호인 경우 숫자랑 *2, *3 곱해서 넣기
4. 스택에 숫자가 연속해서 나오는 경우 숫자 + 숫자 더해서 넣기


상세풀이
1. 여는 괄호 (, [나오면 스택에 쌓기

2. 닫는 괄호 ), ]나오는 경우
2-1 닫는괄호 요청, 길이 0 -> 거짓
2-2 닫는괄호 요청, 스택 꼭대기 여는 괄호  
    2-2.1 여는 괄호 pop
    2-2.2 push 2 || push 3
2-3 닫는괄호 요청, 스택 꼭대기 숫자, 스택 꼭대기 두번째 닫는 괄호
    2-3-1 숫자 pop
    2-3-2 단는 괄호 pop
    2-3-3. push 숫자 * 2  || push 숫자 * 3

3. 스택 꼭대기 숫자, 스택 두번째 꼭대기 숫자
    3-1. 숫자1 pop, 숫자2 pop
    3-2. push 숫자1 + 숫자 2

4. 스택 내부 1개 남은 경우에 숫자인 경우 -> 정답 

 

입력 예시

 

전체코드

s = input()

def solution(s) :
    stack = []
    for i in s :
        #1. 여는 괄호 (, [ 나오면 스택에 쌓기
        if i == "(" or i == "[" :
            stack.append(i)
            continue
        #2. 닫는 괄호 ), ] 나오는 경우
        if i == ")" or i == "]" :
            #2-1. i = 닫는 괄호, 스택이 비어있는 경우 -> 거짓
            if len(stack) == 0 :
                return 0
            
            #2-2. i = 닫는 괄호, 스택 꼭대기가 여는 괄호인 경우
            if stack[-1] == "[" or stack[-1] == "(" : 
                #2-2-1. 닫는 괄호 ) 요청, 스택 꼭대기가 여는 괄호 (인 경우 2 push
                if i == ")" and stack[-1] == "(" :
                    stack.pop()                         #pop
                    stack.append(2)
                #2-2-1. 닫는 괄호 ] 요청, 스택 꼭대기가 여는 괄호 [인 경우 3 push
                elif i == "]" and stack[-1] == "[" :
                    stack.pop()
                    stack.append(3)
            
            #2-3. i = 닫는 괄호, 스택 꼭대기가 숫자인 경우
            else :
                #숫자 뽑자
                num = stack.pop()
                #2-3-1. 닫는 괄호 ) 요청, 숫자 뽑은 다음 스택 꼭대기가 닫는괄호 "("인 경우
                if len(stack) > 0 and i == ")" and stack[-1] == "(" :
                    stack.pop()
                    stack.append(2*num)
                elif len(stack) > 0 and i == "]" and stack[-1] == "[" :
                    stack.pop()
                    stack.append(3*num)
        #3. 스택 안에 연속해서 숫자가 나오는 경우 숫자 두개 팝해서 더한 후 스택에 다시 넣어줌 
        if len(stack) >= 2 and str(stack[-1]).isdigit() and str(stack[-2]).isdigit() :
            num1 = stack.pop()
            num2 = stack.pop()
            stack.append(num1 + num2)
    #4. 스택의 길이가 1이고 스택 내부에 들어 값이 숫자인 경우 반환
    if len(stack) == 1 and str(stack[-1]).isdigit() : 
        return stack[0]
    else :
        return 0

print(solution(s))