Skip to content

课后习题+面试题:用栈和队列实现表达式解析

表达式解析求值,是一个编译器非常重要的能力。本身表达式求值解析,是一个非常复杂的过程。为了将问题简化,思考一个只有整数和加减乘除的表达式,请你使用上节课学习的队列和栈,对它进行求值。

例如:

输入: 1+2 * 3 * 5 + 4 ,得到对应的结果。

然后实现一个交互是的输入控制台,例如:

expr> 1 + 2 * 3 + 4

11

expr> _

解析

可以思考,如果1+2*3+4转化为 1 2 3 * 4 + + 就可以使用栈进行计算。

下面是计算过程,将1 2 3 * 4 + + 依次入栈,如果遇到符号(+-*/)等,就从栈中取出两个元素进行计算,如下图:

image-20210224014138639

那么如何将1 + 2*3 + 4这样的表达式转化为1 2 3 * 4 + +这样的表达式呢?同样需要一个栈。乘号的优先级高,加号优先级低。

遇到高优先级的操作符,应该先入栈,然后遇到低优先级的操作符,就把之前入栈的高优先级操作符出栈,这样就解决了操作符之间的顺序。

最终结果需要一个队列存储,算法还是从左到右处理所有的输入:

image-20210224014456059

1 2 3 4 在队列中的顺序不会发生变化,只不过插入操作符的时机需要把握。

遇到第1个+号时,栈中没有更高优先级的符号,将+号入栈;遇到第1个乘号时,栈中没有更高优先级的符号,将乘号入栈;遇到最后一个+号时,先将乘号出栈,加入队列,再入栈+号,如下图所示:

image-20210224014554388

最终栈中剩余两个+号,都放入队列,就构造了1 2 3 * 4 + + 这个结果。

public class Expr {

    static Queue<String> prefixTransform(String expr) {

        var queue = new Queue<String>();
        var highPriorityStack = new Stack<String>();

        var prts = expr.replaceAll("[+-/*]+", "#$0#")
                .split("#");

        for(var prt : prts) {
            if(prt.matches("\\d+")) {
                queue.enqueue(prt);
            }
            else if(prt.equals("+") || prt.equals("-")) {
                while(highPriorityStack.size() > 0) {
                    var op = highPriorityStack.pop();
                    if(op.equals("*") || op.equals("/")) {
                        queue.enqueue(op);
                    } else {
                        highPriorityStack.push(op);
                        break;
                    }
                }
                highPriorityStack.push(prt);
            } else if(prt.equals("*") || prt.equals("/")) {
                highPriorityStack.push(prt);
            }
        }
        while(highPriorityStack.size() > 0) {
            queue.enqueue(highPriorityStack.pop());
        }
        System.out.println(queue.toString());
        return queue;
    }

    public static Integer eval(String expr) {

        // 1 + 2 * 3 + 4 -> 1 2 3 * 4 + +
        var queue = prefixTransform(expr);
        
        var stack = new Stack<Integer>();
        while(queue.size() > 0) {
            var item = queue.dequeue();
            if(item.matches("\\+|-|\\*|/")) {
                switch (item) {
                    case "+":
                        stack.push(stack.pop() + stack.pop());
                        break;
                    case "-" :
                        stack.push(stack.pop() - stack.pop());
                        break;
                    case "*":
                        stack.push(stack.pop() * stack.pop());
                        break;
                    case "/":
                        stack.push(stack.pop() / stack.pop());
                        break;
                }

            } else {
                stack.push(Integer.parseInt(item));
            }
        }

        return stack.pop();
    }

    public static void main(String[] argv) throws IOException {

        var scanner = new Scanner(System.in);
        while(true) {
            System.out.print("enter>");
            var expr = scanner.nextLine();
            var value = eval(expr);
            System.out.println(value);
        }
    }

}

文章来源于自己总结和网络转载,内容如有任何问题,请大佬斧正!联系我