[Java] 150. Evaluate Reverse Polish Notation

문제 파악

역 폴란드 표기법에 따른 연산 결과를 반환해야한다.

 

연산자를 만나면 이전에 만났던 숫자 2개를 연산하는 방식이다.

 

예로, 4 13 5 / + 의 경우

 

/ 연산자를 만나기 직전 13 과 5가 있으므로

 

13 / 5 를해주고 나온 결과를 다시 넣어주면 4 2 + 가 된다.

 

그리고 + 연산자를 만나기 직전 4와 2가 있으므로 4 + 2 를 한 값인 6을 반환하면 된다.

 

---

풀이

 

1️⃣ 스택을 이용한 풀이

💡 떠오른 Idea

역 폴란드 표기법의 연산 방식 상, 스택을 이용해서 풀어야겠다는 생각이 바로 들었다.

연산자를 만나면, 가장 근처에 있는 수를 뒤로 두고 연산한 다음, 다시 스택에 넣는 방식으로 진행하면 될 것이다.

 

import java.util.*;
class Solution {
    Stack<Integer> stack = new Stack<>();
    Set<String> set = new HashSet<>(List.of("+", "-", "/", "*"));

    public int evalRPN(String[] tokens) {
        for (String token : tokens) {
            if (set.contains(token)) {
                int[] info = getInfo();
                if (token.equals("+")) {
                    stack.push(info[0] + info[1]);
                } else if (token.equals("-")) {
                    stack.push(info[0] - info[1]);
                } else if (token.equals("*")) {
                    stack.push(info[0] * info[1]);
                } else if (token.equals("/")) {
                    stack.push(info[0] / info[1]);
                }
            } else {
                stack.push(Integer.parseInt(token));
            }
        }
        return stack.pop();
    }

    private int[] getInfo() {
        int back = stack.pop();
        int front = stack.pop();
        return new int[]{front, back};
    }
}

 

 결과

 

---

📖 회고

역 폴란드 법 연산이 어떻게 이루어지는지 잘 파악했다면, 스택을 이용해서 쉽게 풀이할 수 있던 문제였다.

 

단, 코드를 짜면서 어떻게 하면 코드의 중복이 없는, 가독성이 좋게 짤 수 있는가를 고민했다

 

따라서 getInfo() 메서드와 set 을 이용한 연산자 구분 조건문을 만들게되었다.