227. 基本计算器 II
给你一个字符串表达式 s
,请你实现一个基本计算器来计算并返回它的值。
整数除法仅保留整数部分。
你可以假设给定的表达式总是有效的。所有中间结果将在 [-2^31, 2^31 - 1]
的范围内。
注意:不允许使用任何将字符串作为数学表达式计算的内置函数,比如 eval()
。
示例 1:
输入:s = "3+2*2"
输出:7
示例 2:
输入:s = " 3/2 "
输出:1
示例 3:
输入:s = " 3+5 / 2 "
输出:5
提示:
1 <= s.length <= 3 * 10^5
s
由整数和算符('+', '-', '*', '/')
组成,中间由一些空格隔开s
表示一个 有效表达式- 表达式中的所有整数都是非负整数,且在范围
[0, 2^31 - 1]
内 - 题目数据保证答案是一个 32-bit 整数
队列+栈
class Solution {
public int calculate(String s) {
Deque<Character> deque = new LinkedList<>();
for (int i = 0; i < s.length(); i++) {
// 入队的时候就把空格排除在外,省的接下来再额外判断。
if (s.charAt(i) != ' ') { // 空格是字符不是字符串,要用单引号
deque.addLast(s.charAt(i));
}
}
return helper(deque);
}
private int helper(Deque<Character> deque) {
Deque<Integer> stack = new LinkedList<>();
char sign = '+';
int num = 0;
while (!deque.isEmpty()) { // 不要忘记 removeFirst 的条件:!deque.isEmpty()
char c = deque.removeFirst();
if (Character.isDigit(c)) { // if 的顺序无所谓。
num = num * 10 + (c - '0'); // c - '0' 是为了把 c 从字符变成数字,和溢出无关
}
if (!Character.isDigit(c) || deque.isEmpty()) {
// 以3+4+5为例,遇到了第二个符号c=+,说明前面的数字已结束,要处理前面的公式。此时的num是4,sign是第一个+,stack.pop是3。计算得7后入栈,此时要把num置为0,不然遇到5时是35而不是5,然后更新sign,下次遇到符号时会用到本轮的sign和下轮的num和本轮入栈的数字。先符号再数字。
// 遇到+-,直接入栈,遇到乘除,先计算,再入栈,最后进行总的求和。
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);
// 每次用完数据之后一定要更新一下数据
if (!deque.isEmpty()) {
num = 0;
sign = c;
}
}
}
// 接下来对栈中的元素求和
int res = 0;
for (Integer i : stack) {
res += i;
}
return res;
}
}
- 时间复杂度:O(n),其中 n 为字符串 s 的长度。需要遍历字符串 s 一次,计算表达式的值。
- 空间复杂度:O(n),其中 n 为字符串 s 的长度。空间复杂度主要取决于栈的空间,栈的元素个数不超过 n。