1、题目
给你一个字符串表达式 s ,请你实现一个基本计算器来计算并返回它的值。
整数除法仅保留整数部分。
你可以假设给定的表达式总是有效的。所有中间结果将在 [-231, 231 - 1] 的范围内。
注意:不允许使用任何将字符串作为数学表达式计算的内置函数,比如 eval() 。
2、思路
整体思路就是遍历字符串的每一位:遇到乘除法就标记,然后运算并压入栈中。为了方便运算,遇到减法将数字的负数压入栈中。最终循环结束统一将数字栈中的结果相加。
这其中涉及到一个问题,就是数字有多位的问题,这里我们用temp进行判断。如果遇到数字后面也是数字,就累加temp,直到后面不是数字,将前面所有位压入栈中。具体遍历过程如下:
1、如果是乘除号,令flag=2,入符号栈。
2、如果是减号,令flag=1,入符号栈,加号默认flag=0
3、如果是数字,且flag=0,直接入栈;
4、如果是数字,且flag=1,该数的负数入栈,flag重置0;
5、如果是数字,且flag=2,和栈顶数字,栈顶符号运算,结果入栈,flag重置0
3、代码
class Solution {
public int calculate(String s) {
Stack<Integer> numStack = new Stack<>();
Stack<Character> strStack = new Stack<>();
int flag = 0; // 用来标记数字前的运算符号
int result = 0;
int temp = 1; // 用来累加多位数字的位数
int len = s.length();
for (int i = 0; i < len; i++) {
char tempChar = s.charAt(i);
if (tempChar == '+') strStack.push(tempChar);
else if (tempChar == '-') {
strStack.push(tempChar);
flag = 1;
} else if (tempChar == '*') {
flag = 2;
strStack.push(tempChar);
} else if (tempChar == '/') {
flag = 2;
strStack.push(tempChar);
} else if (tempChar == ' ') {
} else { //凡是数字都会走到这里
if (flag == 0) { //凡是加号都会走到这里
if (i == len - 1) {// 防止最后一位数字越界
numStack.push(Integer.parseInt(s.substring(i - temp + 1, i + 1)));
temp = 1;
break;
}
char tempCharNext = s.charAt(i + 1);
if (Character.isDigit(tempCharNext)) {// 下一位是数字累加并继续遍历
temp++;
continue;
} else { // 下一位不再是数字就压入栈中
numStack.push(Integer.parseInt(s.substring(i - temp + 1, i + 1)));
temp = 1;
}
} else if (flag == 1) {//凡是减号都会走到这里
if (i == len - 1) {// 防止最后一位数字越界
numStack.push(-Integer.parseInt(s.substring(i - temp + 1, i + 1)));
temp = 1;
break;
}
char tempCharNext = s.charAt(i + 1);
if (Character.isDigit(tempCharNext)) {// 下一位是数字累加并继续遍历
temp++;
continue;
} else {// 下一位不再是数字就压入栈中
numStack.push(-Integer.parseInt(s.substring(i - temp + 1, i + 1)));
temp = 1;
}
} else {//凡是乘除号都会走到这里
if (strStack.peek() == '*') {
if (i == len - 1) {// 防止最后一位数字越界
int num = numStack.pop();
numStack.push(num * Integer.parseInt(s.substring(i - temp + 1, i + 1)));
strStack.pop();
temp = 1;
break;
}
char tempCharNext = s.charAt(i + 1);
if (Character.isDigit(tempCharNext)) {// 下一位是数字累加并继续遍历
temp++;
continue;
} else {// 下一位不再是数字就压入栈中
int num = numStack.pop();
numStack.push(num * Integer.parseInt(s.substring(i - temp + 1, i + 1)));
strStack.pop();
temp = 1;
}
} else {
if (i == len - 1) {// 防止最后一位数字越界
int num = numStack.pop();
numStack.push(num / Integer.parseInt(s.substring(i - temp + 1, i + 1)));
strStack.pop();
temp = 1;
break;
}
char tempCharNext = s.charAt(i + 1);
if (Character.isDigit(tempCharNext)) {// 下一位是数字累加并继续遍历
temp++;
continue;
} else {// 下一位不再是数字就压入栈中
int num = numStack.pop();
numStack.push(num / Integer.parseInt(s.substring(i - temp + 1, i + 1)));
strStack.pop();
temp = 1;
}
}
}
flag = 0;
}
}
if (strStack.isEmpty()) { // 没有符号就不用累加了
return numStack.pop();
}
while (!numStack.isEmpty()) { //累加所有乘除后的结果
result += numStack.pop();
}
return result;
}
}