【Leetcode】150. 逆波兰表达式求值

今天做了一道leetcode，题目不算难，基本就是stack的入栈出栈，但我看了一圈别人的代码，感觉不够优雅，所以我想分享一下自己的写法，主要就是用Map和BiFunction优化4个if else

Java中，函数是能以对象的形式表达。例如(Integer a) - > { return a + 1; }，这就是个自增的函数，我们可以将它赋值给Function increateFunc，也就是

Function<Integer, Integer> increateFunc = (Integer a) -> { return a + 1; };

Function，第一个泛型表示函数入参，第二个泛型表示函数返回类型。

如果函数需要两个入参，可以使用BiFunction

class Solution {
    private static Map<String, BiFunction<Integer, Integer, Integer>> map = new HashMap<>();

    static {
        map.put("+", (a, b) -> {return a + b;});
        map.put("-", (a, b) -> {return a - b;});
        map.put("*", (a, b) -> {return a * b;});
        map.put("/", (a, b) -> {return a / b;});
    }

    public int evalRPN(String[] tokens) {
        Deque<Integer> deque = new ArrayDeque<>();

        for (String token : tokens) {
            if (map.containsKey(token)) {
                Integer num1 = deque.pollLast();
                Integer num2 = deque.pollLast();
                Integer ans = map.get(token).apply(num2, num1);
                deque.addLast(ans);
            }else {
                deque.addLast(Integer.parseInt(token));
            }
        }
        return deque.pollFirst();
    }
}