要求:
1.输入一个逆波兰表达式【后缀表达式】,使用栈(stack)完成计算其结果。
2.支持小括号和多位数。
代码实现
public static void main(String[] args) {
//先定义一个逆波兰表达式
//3 4 + 5 * 6 -
//为了方便,逆波兰表达式的数字和符号用空格隔开
String suffixExpression = "3 4 + 5 * 6 - ";
//思路
//1.先将"3 4 + 5 * 6 - " =》 放到ArrayList中
//2.将ArrayList 传递给一个方法,遍历ArrayList 配合栈完成计算
List<String> rpnList = getListString(suffixExpression);
System.out.println(rpnList);
int res = calculate(rpnList);
System.out.println("计算的结果为:"+res);
}
/**
* 将一个逆波兰表达式,依次将数据和字符放入ArrayList中
*
* @param suffixExpression
* @return
*/
public static List<String> getListString(String suffixExpression) {
//将suffixExpression进行分割
String[] split = suffixExpression.split(" ");
List<String> list = new ArrayList<String>();
for (String a : split) {
list.add(a);
}
return list;
}
/**
* 完成对逆波兰表达式的运算
*/
public static int calculate(List<String> ls) {
//创建一个栈
Stack<String> stack = new Stack<>();
//遍历 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+"");
}
}
//将最后的结果返回
return Integer.parseInt(stack.pop());
}
效果展示:
[3, 4, +, 5, *, 6, -]
计算的结果为:29