题目
给你一个字符串表达式 s ,请你实现一个基本计算器来计算并返回它的值。
代码
class Solution {
public int calculate(String s) {
List<Character> list = new ArrayList<>();
for (int i = 0; i < s.length(); i++) {
list.add(s.charAt(i));
}
return helper(list);
}
int helper(List<Character> s){
Stack<Integer> stk = new Stack<>();
// 记录算式中的数字
int num = 0;
// 记录 num 前的符号,初始化为 +
char sign = '+';
while (s.size()>0) {
char c = s.get(0);//获取字符
s.remove(0);
// 如果是数字,连续读取到 num
if (isdigit(c)) {
num = 10 * num + (c - '0');
}
//遇到左括号开始递归计算 num
if (c=='(') {
num=helper(s);
}
// 如果不是数字,就是遇到了下一个符号,
// 之前的数字和符号就要存进栈中
if ((!isdigit(c) && c != ' ') || s.size() ==0) {
if (sign == '+') {
stk.push(num);
} else if (sign == '-') {
stk.push(-num);
} else if (sign == '*') { // 只要拿出前一个数字做对应运算即可
int pre = stk.pop();
stk.push(pre * num);
} else if (sign == '/') {
int pre = stk.pop();
stk.push(pre / num);
}
//遇到右括号返回递归结果
if (c==')') {
break;
}
// 更新符号为当前符号,数字清零,处理前面的数字
sign = c;
num = 0;
}
}
// 将栈中所有结果求和就是答案
int ans = 0;
while (!stk.isEmpty()) {
ans += stk.pop();
}
return ans;
}
boolean isdigit(char c) {
return c >= '0' && c <= '9';
}
}
要点
- 对加减法,根据符号将数字存入栈中,最后一起取出运算即可
- 对乘除法,由于优先级高,需要将前一个数pop出来进行乘法运算,再将结果压栈
- 对于括号,我们将括号内容视为整体,递归调用将括号内容先计算出来,遇到(开始递归,遇到)结束递归。此时我们用字符长度来判断是否结束。