LeetCode 227. Basic Calculator II
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.
Example 1:
Input: "3+2*2"
Output: 7
Example 2:
Input: " 3/2 "
Output: 1
Example 3:
Input: " 3+5 / 2 "
Output: 5
Note:
You may assume that the given expression is always valid.
Do not use the eval built-in library function.
只有加减乘除,没有括号,因此遇到+, - 都push进栈,遇到*, /就直接运算完再push回栈里。
public int calculate(String s) {
Stack<String> stack = new Stack<>();
for (int i = 0; i < s.length(); i++) {
char c = s.charAt(i);
if (Character.isDigit(c)) {
int[] r = run(s, i);
i = r[1];
if (stack.isEmpty() || stack.peek().equals("+") || stack.peek().equals("-")) {
stack.push(r[0] + "");
} else {
String sign = stack.pop();
int pre = Integer.parseInt(stack.pop());
int result = 0;
result = sign.equals("*") ? pre * r[0] : pre / r[0];
stack.push(result + "");
}
} else if (c != ' ') {
stack.push(c + "");
}
}
int result = 0;
while (!stack.isEmpty()) {
int next = Integer.parseInt(stack.pop());
if (stack.isEmpty()) return (result + next);
String sign = stack.pop();
if (sign.equals("+")) result += next;
else result -= next;
}
return 0;
}
private int[] run(String s, int i) {
int[] res = new int[2];
while (i < s.length() && Character.isDigit(s.charAt(i))) {
res[0] = res[0] * 10 + s.charAt(i++) - '0';
}
res[1] = i - 1;
return res;
}