Implement a basic calculator to evaluate a simple expression string.
The expression string contains only non-negative integers, +
, -
, *
, /
operators and empty spaces . The integer division should truncate toward zero.
You may assume that the given expression is always valid.
Some examples:
"3+2*2" = 7
" 3/2 " = 1
" 3+5 / 2 " = 5
Note: Do not use the eval
built-in library function.
Credits:
Special thanks to @ts for adding this problem and creating all test cases.
思路:这题跟上题不一样的地方在于有了乘法和除法。这样的话,当前的数就要跟前面一个数发生关系,使用了一个stack来保存数字。通过栈来实现运算,按顺序读取字符串,将第一个数入栈。之后遇到+,将下一个数+num入栈;遇到-,则将-num入栈;遇到乘或除,先将上一个数出栈,与当前数进行运算后,再将结果入栈。读取完整个字符串后,将栈中所有的数相加即运算结果。
// 每次是得到当前的运算符,计算之前的数和运算符,然后更新sign为当前的运算符;
// 注意最后一个数字,由于符号总是在数字之前,所以最后一个数字没有办法计算,要单独列出来;
class Solution {
public int calculate(String s) {
if(s == null || s.length() == 0) {
return 0;
}
char sign = '+';
Stack<Integer> stack = new Stack<>();
int num = 0;
int len = s.length();
for(int i = 0; i < len; i++) {
// 取数;
char c = s.charAt(i);
if(Character.isDigit(c)) {
num = num * 10 + c - '0';
}
// 每次是得到当前的运算符,计算之前的数和运算符,然后更新sign为当前的运算符;
// 注意最后一个数字,由于符号总是在数字之前,所以最后一个数字没有办法计算,要单独列出来;
if(!Character.isDigit(c) && c != ' ' || i == len - 1) {
if(sign == '+') {
stack.push(num);
} else if(sign == '-') {
stack.push(-num);
} else if(sign == '*') {
stack.push(stack.pop() * num);
} else if(sign == '/') {
stack.push(stack.pop() / num);
}
sign = c;
num = 0;
}
}
int res = 0;
for(int val : stack) {
res += val;
}
return res;
}
}