[SWEA] 1223. [S/W 문제해결 기본] Stack2 - 계산기2

문제

https://www.swexpertacademy.com/main/learn/course/lectureProblemViewer.do

SW Expert Academy

 

 

풀이

이 문제는 계산기 1 문제와 다르게 연산자가 두 개가 존재한다. 그래서 두 연산자 간의 우선순위를 정해서 후위식으로 변환해줘야 한다. 우선순위 반환 함수를 별도로 만들어주고, 후위식 변환 함수에서 스택에 있는 연산자들의 우선순위에 따라 변환될 수 있도록 구현해주었다.

 

import java.util.*;
 
public class Solution {
    static Scanner scan = new Scanner(System.in);
    static Stack<Character> op = new Stack<Character>();
    static Stack<Integer> sum = new Stack<Integer>();
 
    static int get_priority(char op) {//우선순위 반환 함수
        int pri = 0;
        switch (op) {
        case '+':
            pri = 0;
            break;
        case '*':
            pri = 1;
            break;
        }
        return pri;
    }
 
    static String convert_to_post(String exp) {// 중위 -> 후위
        String post = "";
        for (int i = 0; i < exp.length(); i++) {
            char tmp = exp.charAt(i);
            if (tmp - '0' >= 0 && tmp - '0' < 10)
                post += tmp;
            else {
                while(!op.empty() && get_priority(op.peek())>=get_priority(tmp)) {
                    //우선순위가 *가 더 크기 때문에 스택에 있는 *를 먼저 처리해주기 위한 과정
                    post+=op.pop();
                }
                op.push(tmp);
            }
        }
        while (!op.empty())// 스택에 남아있는 연산자 처리
            post += op.pop();
        return post;
    }
 
    static int calc_post_exp(String exp) {// 후위식 계산
        int answer = 0;
        int n1, n2;
        for (int i = 0; i < exp.length(); i++) {
            char tmp = exp.charAt(i);
            if (tmp - '0' >= 0 && tmp - '0' < 10)
                sum.push(tmp - '0');
            else {
                switch (tmp) {
                case '+': {
                    int result = 0;
                    for (int j = 0; j < 2; j++)
                        result += sum.pop();
                    sum.push(result);
                    break;
                }
                case '*': {
                    int result = 1;
                    for (int j = 0; j < 2; j++)
                        result *= sum.pop();
                    sum.push(result);
                    break;
                }
                }
            }
        }
        answer = sum.pop();
        return answer;
    }
 
    static void solution() {
        String str = "";
        for (int i = 1; i <= 10; i++) {
            scan.nextInt();
            str = scan.next();
            System.out.println("#" + i + " " + calc_post_exp(convert_to_post(str)));
        }
    }
 
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        solution();
    }
}