LeetCode刷题最好按照标签归类来刷,本文主要对leetcode上的表达式求值(基本计算器)问题做一个梳理。
这个问题实际上是一个很基础也很常见的问题,在《算法4》中也有提及使用双栈来解决,本文基于双栈求解该问题做了一个详解。
实际在上一篇LeetCode 括号问题专栏中也提及了栈结构可用于处理表达式求值问题,在LeetCode中该类问题也有很多,理解本文后,可以使用该解法解决LC上所有该类问题。
- LC224. 基本计算器
- LC227. 基本计算器 II
- LC772. 基本计算器 III
基本计算器核心内容
双栈,指的是一个用于存储操作数的数字栈,一个用于存储操作符的符号栈。
定义一个计算过程:
在计算的过程中,大概过程是这样:从符号栈中取出一个符号(出栈),从数字栈栈中取出两个数(出栈),进行计算,得到的结果再存入数字栈中,直到符号栈为空。
eg:下面这个表达式如何求解计算结果呢
String str = " -1+( -1 * 6 - 8 )*(-2)"
首先我们先分析表达式的组成:主要有3个部分(空格、数字、操作符)组成,同时我们对于这三类字符就会有相应的处理。
- 空格符号
对于空格符号,在遍历过程中直接跳过即可 - 数字
当我们遍历到一个数字时,需要获取完整的数字串,作为一个操作数去进行计算。因此需要判断后续字符是否是数字,如果是的话,按照十进制加权求和进行计算。
num = num*10+a[i] - 操作符(包括加减乘除,还有左右括号)
1.对于左括号,直接压入操作符栈
2.对于右括号,说明此时要进行结算了。按照上述定义的计算过程对两个栈进行操作,直到符号栈的栈顶元素为’(’,此时说明该右括号的作用范围结束了。
3.对于 加减乘除 符号,需要根据符号的优先级判断此时是否能够进行计算
1)如果此时符号栈为空,直接将 操作符 压入栈中;
2)符号栈不为空,当前栈顶元素的操作符的优先级 大于或者等于 当前遍历到的
操作符的优先级,那么可以循环进行上述定义的计算过程,直到符号栈为空或者
栈顶元素是左括号
需要考虑的一个细节是,一个操作符一般是对应于两个操作数,但是存在这样一种情况,操作数是作为数字的正负符号的,如例子中开头的-1和结尾括号处的(-2)。这种case会影响我们正常的程序运行,因此我们可以把它转换成一个操作符对应两个操作数,在这种case前面加上0。那这种case是如何识别的呢,其实就是 当前是一个操作符,如果它的前一个非空字符是左括号,加号,减号,那么就给该操作符前,数字栈先添加0。
下面为具体的代码:
class Solution {
public int calculate(String s) {
char[] array = s.toCharArray();
//操作符栈
Stack<Character> operatorStack = new Stack<>();
// 数字栈
Stack<Integer> numStack = new Stack<>();
// 第一个数为负数,先往 nums 加个 0
numStack.push(0);
// 记录上一个不为空的字符
int lastNotBlank = -1;
for (int i = 0; i < array.length; i++) {
char ch = array[i];
//空格过滤掉