今天做了一道leetcode,题目不算难,基本就是stack的入栈出栈,但我看了一圈别人的代码,感觉不够优雅,所以我想分享一下自己的写法,主要就是用Map和BiFunction优化4个if else
Java中,函数是能以对象的形式表达。例如(Integer a) - > { return a + 1; }
,这就是个自增的函数,我们可以将它赋值给Function<Integer, Integer> increateFunc
,也就是
Function<Integer, Integer> increateFunc = (Integer a) -> { return a + 1; };
Function<Integer, Integer>,第一个泛型表示函数入参,第二个泛型表示函数返回类型。
如果函数需要两个入参,可以使用BiFunction<T, U, R>
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();}
}