LeetCode 150 Evaluate Reverse Polish Notation

题目描述

Evaluate the value of an arithmetic expression in Reverse Polish Notation.

Valid operators are +, -, *, /. Each operand may be an integer or another expression.

Some examples:

["2", "1", "+", "3", "*"] -> ((2 + 1) * 3) -> 9
["4", "13", "5", "/", "+"] -> (4 + (13 / 5)) -> 6

分析

这道题是给定了合法的逆波兰表达式，要求根据给定的逆波兰表达式，求出最终结果。

只需要定义一个stack，如果是+, -, *, /四个运算符，就取出栈顶的两个数，进行相应操作。之后将计算得到的结果压入栈中。如果是数字，就直接入栈。

最终stack只剩下一个元素，这个元素就是逆波兰表达式的值。

逆波兰表达式

逆波兰表达式又叫做后缀表达式。下面是一些例子：

正常的表达式 逆波兰表达式
a+b —> a,b,+
a+(b-c) —> a,b,c,-,+
a+(b-c)d —> a,b,c,-,d,,+
a+d*(b-c)—>a,d,b,c,-,*,+
a=1+3 —> a=1,3 +

优势

它的优势在于只用两种简单操作，入栈和出栈就可以搞定任何普通表达式的运算。其运算方式如下：

如果当前字符为变量或者为数字，则压栈，如果是运算符，则将栈顶两个元素弹出作相应运算，结果再入栈，最后当表达式扫描完后，栈里的就是结果。

代码

    public static int evalRPN(String[] tokens) {

        Stack<Integer> stack = new Stack<Integer>();

        for (String s : tokens) {

            if ("*".equals(s)) {
                int n2 = stack.pop();
                int n1 = stack.pop();

                stack.push(n1 * n2);
            } else if ("/".equals(s)) {
                int n2 = stack.pop();
                int n1 = stack.pop();

                stack.push(n1 / n2);
            } else if ("+".equals(s)) {
                int n2 = stack.pop();
                int n1 = stack.pop();

                stack.push(n1 + n2);
            } else if ("-".equals(s)) {
                int n2 = stack.pop();
                int n1 = stack.pop();

                stack.push(n1 - n2);
            } else {
                stack.push(Integer.valueOf(s));
            }

        }

        return stack.pop();
    }