JAVA代码实现中缀表达式转后缀表达式并求值

中缀表达式转后缀表达式并求值

最近学习了利用栈实现中缀表达式转后缀表达式并求值的实现方式。目前只实现了加减乘除操作，小数部分暂时未考虑。如有错误之处，还望指出，我会及时改正。

package com.stack.calulator;

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

/*
 * 一、将中缀表达式转成后缀表达式(逆波兰表达式)：
 *    步骤分析：
 *          1.初始化两个栈，运算符栈s1和数栈s2;
            2.从左至右扫描中缀表达式:
            3.遇到数时，直接压入s2;
            4.遇到运算符时，比较s1栈顶运算符的优先级；
               4.1.如果s1为空，或者栈顶运算符为左括号，则直接将此运算符入s1；
               4.2.如果优先级比栈顶运算符高，直接将此运算符压入s1;
               4.3.否则将s1栈顶的运算符弹出并压入s2，在此转到4.1与新的栈顶运算符相比较；
            5.遇到括号时：
               5.1.如果是左括号，直接入s1;
               5.2.如果是右括号，依次弹出s1栈顶的运算符，直到遇到左括号，此时这一对括号丢  掉；
            6.重复步骤2至5，直到表达式的最右边；
            7.将s1中剩余的运算符依次弹出并压入s2；
            8.依次弹出s2中的元素，结果的逆序就是对应的后缀表达式
 *
 *二、计算后缀表达式
 *   步骤分析：
 *         1.从左至右扫描后缀表达式；
 *         2.遇到数直接压入栈；
 *         3.遇到运算符时，从栈中取出两个数进行运算，结果压入栈；
 *         4.扫描结束时，栈里的那个数就是结果
 */
public class PolandNotationDemo2 {

	public static void main(String[] args) {
		// 定义一个字符串
		String expression = "1+((2+3)*4)-5";
		// 将字符串转成中缀表达式存放到list中
		List<String> infixExpressionList = toInfixExpressionList(expression);
		System.out.println(infixExpressionList);
		// 将中缀表达式转换成后缀表达式
		List<String> postfixExpressionList = toPostfixExpressionList(infixExpressionList);
		System.out.println(postfixExpressionList);
		// 计算后缀表达式
		calculatePostfixExpression(postfixExpressionList);
	}

	/**
	 * 将字符串转成中缀表达式存放到list中
	 * 
	 * @param expression
	 * @return
	 */
	public static List<String> toInfixExpressionList(String expression) {
		List<String> list = new ArrayList<String>();
		char ch;// 存放每次遍历的字符
		String str;// 多位数拼接使用
		int index = 0;// 指针
		do {
			if ((ch = expression.charAt(index)) < 48 || (ch = expression.charAt(index)) > 57) {// 如果是符号，直接添加
				list.add("" + ch);
				index++;
			} else {// 如果是多位数
				str = "";// 每次进来先置空
				while (index < expression.length() && (ch = expression.charAt(index)) >= 48
						&& (ch = expression.charAt(index)) <= 57) {
					str += ch;
					index++;
				}
				list.add(str);
			}
		} while (index < expression.length());

		return list;
	}

	/**
	 * 中缀表达式转成后缀表达式
	 * 
	 * @param infixExpressionList
	 * @return
	 */
	public static List<String> toPostfixExpressionList(List<String> infixExpressionList) {
		/*
		 * 1.初始化两个栈，运算符栈s1和数栈s2;
		   2.从左至右扫描中缀表达式:
		   3.遇到数时，直接压入s2;
		   4.遇到括号时：
		       4.1.如果是左括号，直接入s1;
		       4.2.如果是右括号，依次弹出s1栈顶的运算符，压入s2,直到遇到左括号，此时这一对括号丢 掉；
		   5.遇到运算符时，比较s1栈顶运算符的优先级；
		       5.1.如果s1为空，或者栈顶运算符为左括号，则直接将此运算符入s1；
		       5.2.如果优先级比栈顶运算符高，直接将此运算符压入s1;
		       5.3.否则将s1栈顶的运算符弹出并压入s2，在此转到4.1与新的栈顶运算符相比较；
		   6.重复步骤2至5，直到表达式的最右边；
		   7.将s1中剩余的运算符依次弹出并压入s2；
		   8.依次弹出s2中的元素，结果的逆序就是对应的后缀表达式
		 */

		// 1.初始化符号栈和数栈
		Stack<String> s1 = new Stack<String>();
		// 因为数栈只涉及存放数据，如果使用栈最后还要逆序输出，所以直接使用ArrayList代替
		List<String> s2 = new ArrayList<String>();
		// 2.遍历中缀表达式
		for (String item : infixExpressionList) {
			// 3.遇到数时，直接压入栈
			if (item.matches("\\d+")) {// 正则表达式判断多位数
				s2.add(item);
			} else if (item.equals("(")) {// 如果是左括号，直接入s1
				s1.push(item);
			} else if (item.equals(")")) {// 如果是右括号
				// 依次弹出s1栈顶的运算符，压入s2,直到遇到左括号
				while (!s1.peek().equals("(")) {
					s2.add(s1.pop());
				}
				// 碰到左括号跳出循环了，将这个左括号去掉
				s1.pop();
			} else {// 遇到运算符
					// 如果s1栈为空或者栈顶元素是(,直接入栈
				if (s1.isEmpty() || s1.peek().equals("(")) {
					s1.push(item);
				} else if (priority(item) > priority(s1.peek())) {// 如果优先级大于栈顶优先级，直接入s1
					s1.push(item);
				} else {// 如果优先级小于等于栈顶优先级，将s1栈顶的运算符弹出并压入s2，再与新的栈顶运算符相比较
					while (s1.size() > 0 && priority(s1.peek()) >= priority(item)) {
						s2.add(s1.pop());
					}
					// 循环结束，再将item压入s1s
					s1.push(item);
				}
			}
		}
		// 将s1中剩余的运算符依次弹出并压入s2
		while (s1.size() > 0) {
			s2.add(s1.pop());
		}
		return s2;
	}

	/**
	 * 计算后缀表达式
	 * 
	 * @param list
	 */
	public static void calculatePostfixExpression(List<String> list) {
		/*
		 * 1.从左至右扫描后缀表达式；
		 * 2.遇到数直接压入栈；
		 * 3.遇到运算符时，从栈中取出两个数进行运算，结果压入栈；
		 * 4.扫描结束时，栈里的那个数就是结果
		 */

		Stack<Integer> stack = new Stack<Integer>();
		int num1 = 0;
		int num2 = 0;
		int res = 0;
		for (String item : list) {
			if (item.matches("\\d+")) {// 如果是数，直接入栈
				stack.push(Integer.parseInt(item));
			} else {// 如果是运算符，从栈中取出两个数进行运算，结果压入栈
				num1 = stack.pop();
				num2 = stack.pop();
				res = calculate(num1, num2, item);
				stack.push(res);
			}
		}
		System.out.printf("计算结果为：%d",stack.pop());
	}

	/**
	 * 计算方式
	 * 
	 * @param num1
	 * @param num2
	 * @param oper
	 * @return
	 */
	public static int calculate(int num1, int num2, String oper) {
		int res = 0;
		if (oper.equals("+")) {
			res = num2 + num1;
		}
		if (oper.equals("-")) {
			res = num2 - num1;
		}
		if (oper.equals("*")) {
			res = num2 * num1;
		}
		if (oper.equals("/")) {
			res = num2 / num1;
		}
		return res;
	}

	/**
	 * 根据传入的运算符比较优先级，优先级越高值越大
	 * 
	 * @param oper
	 * @return
	 */
	public static int priority(String oper) {
		int res = 0;
		if (oper.equals("+") || oper.equals("-")) {
			res = 1;
		}
		if (oper.equals("*") || oper.equals("/")) {
			res = 2;
		}
		return res;
	}

}