4.7-逆波兰计算(后缀表达式):完成一个逆波兰计算器,分析和代码实现

4.7-逆波兰计算(后缀表达式):完成一个逆波兰计算器,分析和代码实现

• 要求完成以下任务
  1. 输入一个逆波兰表达式(后缀表达式),使用栈,计算其结果
  2. 支持小括号和多位整数
  3. 思路分析
  4. 代码完成

package com.hejiale.stack.v4;

import java.util.ArrayList;
import java.util.List;
import java.util.Stack;

public class PolandNotation {
    public static void main(String[] args) {
        //先定义一个逆波兰表达式,为了方便,数字和符号使用空格隔开
        String suffixExpression = "30 4 + 5 * 6 - ";
        /*
            思路
                1.  先将suffixExpression装入到一个容器ArrayList中
                2.  将ArrayList传给一个方法,这个方法配合栈,完成计算
        */
        List<String> listString = getListString(suffixExpression);
        int count = count(listString);
        System.out.println(count);
    }

    /**
     * 传入一个逆波兰表达式,这个表达式用空格分割,然后将逆波兰表达式分割后装入一个列表并返回
     * @param suffixExpression 逆波兰表达式
     * @return 将逆波兰表达式装入列表并返回
     */
    public static List<String> getListString (String suffixExpression){
        // 将suffixExpression分割
        String[] s = suffixExpression.split(" ");
        List<String> list = new ArrayList<String>();
        for (String s1 : s) {
            list.add(s1);
        }
        return list;
    }

    /**
     * 传入一个波兰表达式,该方法会计算出结果并返回
     * @param list 波兰表达式
     * @return 表达式的结果
     */
    public static int count(List<String> list){
        Stack<String> stack = new Stack<>();
        for (String item : list) {
            //使用正则表达式取出数
            if(item.matches("\\d+")){//匹配的是多位数
                //入栈
                stack.push(item);
            }else {
                //弹出两个数,并运算,再入栈
                int num2 = Integer.parseInt(stack.pop());//栈顶元素
                int num1 = Integer.parseInt(stack.pop());//次顶元素
                int res = 0;
                if(item.equals("+")){
                    res = num1 + num2;
                } else if (item.equals("-")){
                    res = num1 - num2;
                } else if (item.equals("*")) {
                    res = num1 * num2;
                } else if (item.equals("/")){
                    res = num1 / num2;
                } else {
                    throw new RuntimeException("逆波兰表达式不正确");
                }
                // 将res入栈
                stack.push(res + "");
            }
        }

        //最后留在stack中的数字就是结果
        return Integer.parseInt(stack.pop());
    }
}