尚硅谷Java数据结构与算法——037逆波兰计算器

代码实现

package com.atguigu.stack;

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

/**
 * @author WHT
 * @create 2021-06-11 9:22
 */
public class PolandNotation {

    public static void main(String[] args) {

        // 先定义给逆波兰表达式
        // (30+4)×5-6  => 30 4 + 5 × 6 - => 164
        // 4 * 5 - 8 + 60 + 8 / 2 => 4 5 * 8 - 60 + 8 2 / +
        // 测试
        // 说明为了方便，逆波兰表达式 的数字和符号使用空格隔开。
        String suffixExpression = "4 5 * 8 - 60 + 8 2 / +";

        //思路
        //1. 先将 "3 4 + 5 × 6 - " => 放到ArrayList中；
        //2. 将 ArrayList 传递给一个方法，遍历 ArrayList 配合栈 完成计算。
        List<String> rpnList = getListString(suffixExpression);
        System.out.println("rpnList=" + rpnList);

        // 测试 calculate 。
        int res = calculate(rpnList);
        System.out.println("计算结果=" + res);
    }

    // 将一个逆波兰表达式， 依次将数据和运算符 放入到 ArrayList中。
    public static List<String> getListString(String suffixExpression) {
        // 将 suffixExpression 分割。
        String[] split = suffixExpression.split(" ");
        List<String> list = new ArrayList<String>();
        for (String ele: split) {
            list.add(ele);
        }
        return list;
    }

    // 完成对逆波兰表达式的运算。
	/*
	 * 1)从左至右扫描，将3和4压入堆栈；
		2)遇到+运算符，因此弹出4和3（4为栈顶元素，3为次顶元素），计算出3+4的值，得7，再将7入栈；
		3)将5入栈；
		4)接下来是×运算符，因此弹出5和7，计算出7×5=35，将35入栈；
		5)将6入栈；
		6)最后是-运算符，计算出35-6的值，即29，由此得出最终结果。
	 */
    public static int calculate(List<String> ls) {
        // 创建栈, 只需要一个栈即可。
        Stack<String> stack = new Stack<String>();
        // 遍历 ls。
        for (String item: ls) {
            // 这里使用正则表达式来取出数。
            if (item.matches("\\d+")) { // 匹配的是多位数。
                // 入栈。
                stack.push(item);
            }
            else {
                // pop出两个数，并运算， 再入栈。
                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());
    }
}

运行结果

D:\java\jdk-11.0.11\bin\java.exe "-javaagent:D:\IntelliJ IDEA 2020.1.1\lib\idea_rt.jar=6923:D:\IntelliJ IDEA 2020.1.1\bin" -Dfile.encoding=UTF-8 -classpath E:\WHT\Project\Java\DataStructures\out\production\DataStructures com.atguigu.stack.PolandNotation
rpnList=[4, 5, *, 8, -, 60, +, 8, 2, /, +]
计算结果=76

Process finished with exit code 0