[백준 2504] 괄호의 값

[문제]

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

한 쌍의 괄호로만 이루어진 ‘()’와 ‘[]’는 올바른 괄호열이다.
만일 X가 올바른 괄호열이면 ‘(X)’이나 ‘[X]’도 모두 올바른 괄호열이 된다.
X와 Y 모두 올바른 괄호열이라면 이들을 결합한 XY도 올바른 괄호열이 된다.
예를 들어 ‘(()[[]])’나 ‘(())[][]’ 는 올바른 괄호열이지만 ‘([)]’ 나 ‘(()()[]’ 은 모두 올바른 괄호열이 아니다. 우리는 어떤 올바른 괄호열 X에 대하여 그 괄호열의 값(괄호값)을 아래와 같이 정의하고 값(X)로 표시한다.

‘()’ 인 괄호열의 값은 2이다.
‘[]’ 인 괄호열의 값은 3이다.
‘(X)’ 의 괄호값은 2×값(X) 으로 계산된다.
‘[X]’ 의 괄호값은 3×값(X) 으로 계산된다.
올바른 괄호열 X와 Y가 결합된 XY의 괄호값은 값(XY)= 값(X)+값(Y) 로 계산된다.
예를 들어 ‘(()[[]])([])’ 의 괄호값을 구해보자. ‘()[[]]’ 의 괄호값이 2 + 3×3=11 이므로 ‘(()[[]])’의 괄호값은 2×11=22 이다. 그리고 ‘([])’의 값은 2×3=6 이므로 전체 괄호열의 값은 22 + 6 = 28 이다.

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

[입력]

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

[출력]

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

[풀이]

스택 자료구조로 풀 수 있는 문제이다. 여는 괄호가 나오면 스택에 넣고 소괄호는 tmp 값에 X2를 하고, 대괄호는 X3을 한다. 닫는 괄호가 나오면 스택이 비어있거나 스택 맨 위 값이 같은 형태의 여는 괄호가 아니면 result에 0을 넣고 for문을 나간다. 현재문자에서 전 문자가 같은 형태의 여는 괄호라면 현재 tmp값을 result에 더해주고 더해준 값 만큼 다시 tmp값을 2 or 3으로 나눠준다.

[코드]

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;

public class Main {

    static Stack<Character> st;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        String s = br.readLine();

        st = new Stack<>();

        int result = 0;
        int tmp = 1;

        for(int i=0; i<s.length(); i++){
            if(s.charAt(i) == '('){         //소괄호 열기
                st.push(s.charAt(i));
                tmp *= 2;
            }
            else if(s.charAt(i) == ')'){    //소괄호 닫기
                if(st.isEmpty() || st.peek() != '('){
                    result = 0;
                    break;
                }
                else if(s.charAt(i-1) == '('){
                    result += tmp;
                }
                st.pop();
                tmp /=2 ;
            }
            else if(s.charAt(i) == '['){    //대괄호 열기
                st.push(s.charAt(i));
                tmp *= 3;
            }
            else {                          //대괄호 닫기
                if(st.isEmpty() || st.peek() != '['){
                    result = 0;
                    break;
                }
                else if(s.charAt(i-1) == '['){
                    result += tmp;
                }
                st.pop();
                tmp /= 3;
            }
        }
        if(!st.isEmpty()) System.out.println(0);
        else System.out.println(result);
    }
}