Java数据结构与算法———（24）用Java包中的栈计算逆波兰表达式（后缀表达式）

用Java包中的栈来计算逆波兰表达式，也即后缀表达式。

一、代码

package stack;

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

public class PolandNotation {

	public static void main(String[] args) {
		// 先定义一个逆波兰表达式（字符串）
		// (3+4)x5-6 => 3 4 + 5 x 6 -
		// 为了方便，逆波兰表达式的数字和符号之间要用空格隔开
		String suffixExpression = "3 4 + 5 * 6 -";

		// 思路
		// 1. 先将"3 4 + 5 x 6 - "放到 ArrayList 中
		// 2. 将 ArrayList 传递给一个方法，遍历 ArrayList 配合栈完成计算

		List<String> list = getListString(suffixExpression);
		System.out.println("rpnList = " + list);
		try {
			int res = cal(list);
			System.out.println("计算结果 = " + res);
		} catch (Exception e) {
			System.out.println(e.getMessage());
		}
	}

	// 依次将一个逆波兰表达式的数字和运算符放入到ArrayList中
	public static List<String> getListString(String suffixExpression) {
		// 将字符串 suffixExpression 分割成单个字符串（不包括空格字符串" "），然后放入一个字符串数组
		// suffixExpression.split(" ")表示以空格字符串" "作为分隔符将suffixExpression分割成单个的字符串
		String[] split = suffixExpression.split(" ");
		List<String> list = new ArrayList<String>();
		for (String e : split) {
			list.add(e);
		}
		return list;
	}

	// 完成对逆波兰表达式 3 4 + 5 x 6 - 的运算
	// 1.从左至右扫描，将3和4压入堆栈；
	// 2.遇到+运算符，因此弹出4和3（4为栈顶元素，3为次顶元素），计算出3+4=7，再将7入栈；
	// 3.将5入栈；
	// 4.接下来是x运算符，因此弹出5和7，计算7x5=35，将25入栈； 5)将6入栈； 6)最后是-运算符，计算出35-6=29，由此得出最终结果。

	public static int cal(List<String> ls) {
		// 创建一个栈，只需要一个栈即可
		Stack<String> stack = new Stack<String>();
		// 遍历传入的ls
		for (String e : ls) {
			// 这里使用正则表达式取出数
			if (e.matches("\\d+")) {// 匹配的是多位数
				stack.push(e);// 如果e是一个数，则直接入栈
			} else {
				// 如果e不是数，则pop出两个数，并运算，再入栈
				// 注意：先弹出的数用num2接收，后弹出的数用num1接收
				int num2 = Integer.parseInt(stack.pop());// 出栈的字符串，所以要转换成整型数值
				int num1 = Integer.parseInt(stack.pop());
				int res = 0;// 用于存放num2和num1的计算结果
				if (e.equals("+")) {// 如果扫描到运算符是+
					res = num1 + num2;// 注意：先弹出的数用num2接收，后弹出的数用num1接收
				} else if (e.equals("-")) {
					// 注意：先弹出的数用num2接收，后弹出的数用num1接收，所以res = num1 - num2
					// 如果先弹出的数用num1接收，后弹出的数用num2接收，则res = num2 - num1
					res = num1 - num2;
				} else if (e.equals("*")) {
					res = num1 * num2;
				} else if (e.equals("/")) {
					// 注意：先弹出的数用num2接收，后弹出的数用num1接收，所以res = num1/num2
					// 如果先弹出的数用num1接收，后弹出的数用num2接收，则res = num2/num1
					res = num1 / num2;
				} else {// 如果运算符不是+、-、*、/中的一个，则抛出异常
					throw new RuntimeException("运算符有误");
				}
				// 把运算结果res入栈
				stack.push("" + res);// 栈stack上一个字符串栈，而res是整数，所以要将res变成"res"才能入栈
			}
		}
		// 最后留在stack中的数据就是运算结果
		// 注意：stack.pop()的结果是一个字符串，要将字符串转换成int
		return Integer.parseInt(stack.pop());
	}
}

二、结果

rpnList = [3, 4, +, 5, *, 6, -]
计算结果 = 29