本文正在参加「金石计划 . 瓜分6万现金大奖」
表达式计算问题
关于「表达式计算问题」一直是笔试面试中较难的部分,所幸的时该做法有高度通用的做法:双栈做法。
本文将大家使用「双栈」做法解决 到表达式计算相关问题,同时对于涉及需要数学证明的地方都会有完备证明。
592. 分数加减运算
给定一个表示分数加减运算的字符串 expression,你需要返回一个字符串形式的计算结果。
这个结果应该是不可约分的分数,即最简分数。 如果最终结果是一个整数,例如 ,你需要将它转换成分数形式,其分母为 。所以在上述例子中, 应该被转换为 2/1。
示例 1:
输入: expression = "-1/2+1/2"
输出: "0/1"
示例 2:
输入: expression = "-1/2+1/2+1/3"
输出: "1/3"
示例 3:
输入: expression = "1/3-1/2"
输出: "-1/6"
提示:
- 输入和输出字符串只包含
'0'到'9'的数字,以及'/','+'和'-'。 - 输入和输出分数格式均为
±分子/分母。如果输入的第一个分数或者输出的分数是正数,则'+'会被省略掉。 - 输入只包含合法的最简分数,每个分数的分子与分母的范围是 。 如果分母是 ,意味着这个分数实际上是一个整数。
- 输入的分数个数范围是 。
- 最终结果的分子与分母保证是 位整数范围内的有效整数。
表达式计算
为了方便,令 expression 为 s。
由于给定的表达式中只有 + 和 -,因此无须考虑优先级问题,直接从前往后计算即可。
使用变量 ans 代指计算过程中的结果,从前往后处理表达式 s,每次以 ±分子/分母 的形式取出当前操作数(若为表达式的首个操作数,且为正数时,需要手动补一个 +)。
假设当前取出的操作数为 num,根据 ans 的情况进行运算:
- 若
ans为空串,说明num是首个操作数,直接将num赋值给ans即可 - 若
ans不为空串,此时计算num和ans的计算结果赋值给ans
考虑实现一个计算函数 String calc(String a, String b) 计算两个操作 a 和 b 的结果,其中入参 a 和 b 以及返回值均满足 ±分子/分母 形式。
首先通过读取 a 和 b 的首个字符,得到两操作数的正负情况。若为一正一负,通过交换的形式,确保 a 为正数,b 为负数。
然后通过 parse 方法拆解出字符串操作数的分子和分母,parse 使用指针扫描的方式实现即可,以数组形式将结果返回(第 位为分子数值,第 位分母数值)。
假设操作数 a 对应的值为 ,操作数的值为 ,先将其转换为 和 ,进行运算后,再通过求最大公约数 gcd 的方式进行化简。
Java 代码:
class Solution {
public String fractionAddition(String s) {
int n = s.length();
char[] cs = s.toCharArray();
String ans = "";
for (int i = 0; i < n; ) {
int j = i + 1;
while (j < n && cs[j] != '+' && cs[j] != '-') j++;
String num = s.substring(i, j);
if (cs[i] != '+' && cs[i] != '-') num = "+" + num;
if (!ans.equals("")) ans = calc(num, ans);
else ans = num;
i = j;
}
return ans.charAt(0) == '+' ? ans.substring(1) : ans;
}
String calc(String a, String b) {
boolean fa = a.charAt(0) == '+', fb = b.charAt(0) == '+';
if (!fa && fb) return calc(b, a);
long[] p = parse(a), q = parse(b);
long p1 = p[0] * q[1], q1 = q[0] * p[1];
if (fa && fb) {
long r1 = p1 + q1, r2 = p[1] * q[1], c = gcd(r1, r2);
return "+" + (r1 / c) + "/" + (r2 / c);
} else if (!fa && !fb) {
long r1 = p1 + q1, r2 = p[1] * q[1], c = gcd(r1, r2);
return "-" + (r1 / c) + "/" + (r2 / c);
} else {
long r1 = p1 - q1, r2 = p[1] * q[1], c = gcd(Math.abs(r1), r2);
String ans = (r1 / c) + "/" + (r2 / c);
if (p1 >= q1) ans = "+" + ans;
return ans;
}
}
long[] parse(String s) {
int n = s.length(), idx = 1;
while (idx < n && s.charAt(idx) != '/') idx++;
long a = Long.parseLong(s.substring(1, idx)), b = Long.parseLong(s.substring(idx + 1));
return new long[]{a, b};
}
long gcd(long a, long b) {
return b == 0 ? a : gcd(b, a % b);
}
}
TypeScript 代码:
function fractionAddition(s: string): string {
const n = s.length
let ans = ""
for (let i = 0; i < n; ) {
let j = i + 1
while (j < n && s[j] != '+' && s[j] != '-') j++
let num = s.substring(i, j)
if (s[i] != '+' && s[i] != '-') num = "+" + num
if (ans != "") ans = calc(num, ans)
else ans = num
i = j
}
return ans[0] == "+" ? ans.substring(1) : ans
};
function calc(a: string, b: string): string {
const fa = a[0] == "+", fb = b[0] == "+"
if (!fa && fb) return calc(b, a)
const p = parse(a), q = parse(b)
const p1 = p[0] * q[1], q1 = q[0] * p[1]
if (fa && fb) {
const r1 = p1 + q1, r2 = p[1] * q[1], c = gcd(r1, r2)
return "+" + (r1 / c) + "/" + (r2 / c)
} else if (!fa && !fb) {
const r1 = p1 + q1, r2 = p[1] * q[1], c = gcd(r1, r2)
return "-" + (r1 / c) + "/" + (r2 / c)
} else {
const r1 = p1 - q1, r2 = p[1] * q[1], c = gcd(Math.abs(r1), r2)
let ans = (r1 / c) + "/" + (r2 / c)
if (p1 > q1) ans = "+" + ans
return ans
}
}
function parse(s: string): number[] {
let n = s.length, idx = 1
while (idx < n && s[idx] != "/") idx++
const a = Number(s.substring(1, idx)), b = Number(s.substring(idx + 1))
return [a, b]
}
function gcd(a: number, b: number): number {
return b == 0 ? a : gcd(b, a % b)
}
- 时间复杂度:
- 空间复杂度:
1006. 笨阶乘
通常,正整数 n 的阶乘是所有小于或等于 n 的正整数的乘积。
例如,factorial(10) = 10 * 9 * 8 * 7 * 6 * 5 * 4 * 3 * 2 * 1。
相反,我们设计了一个笨阶乘 clumsy:在整数的递减序列中,我们以一个固定顺序的操作符序列来依次替换原有的乘法操作符:乘法(*),除法(/),加法(+)和减法(-)。
例如,clumsy(10) = 10 * 9 / 8 + 7 - 6 * 5 / 4 + 3 - 2 * 1。然而,这些运算仍然使用通常的算术运算顺序:我们在任何加、减步骤之前执行所有的乘法和除法步骤,并且按从左到右处理乘法和除法步骤。
另外,我们使用的除法是地板除法(floor division),所以 10 * 9 / 8 等于 11。这保证结果是一个整数。
实现上面定义的笨函数:给定一个整数 N,它返回 N 的笨阶乘。
示例 1:
输入:4
输出:7
解释:7 = 4 * 3 / 2 + 1
示例 2:
输入:10
输出:12
解释:12 = 10 * 9 / 8 + 7 - 6 * 5 / 4 + 3 - 2 * 1
提示:
- 1 <= N <= 10000
- - <= answer <= - 1 (答案保证符合 32 位整数)
通用表达式解法
第一种解法是我们的老朋友解法了,使用「双栈」来解决通用表达式问题。
事实上,我提供这套解决方案不仅仅能解决只有 + - ( )(224. 基本计算器) 或者 + - * /(227. 基本计算器 II) 的表达式问题,还能能解决 + - * / ^ % ( ) 的完全表达式问题。
甚至支持自定义运算符,只要在运算优先级上进行维护即可。
对于「表达式计算」这一类问题,你都可以使用这套思路进行解决。我十分建议你加强理解这套处理逻辑。
对于「任何表达式」而言,我们都使用两个栈 nums 和 ops:
nums: 存放所有的数字ops:存放所有的数字以外的操作
然后从前往后做,对遍历到的字符做分情况讨论:
- 空格 : 跳过
(: 直接加入ops中,等待与之匹配的)): 使用现有的nums和ops进行计算,直到遇到左边最近的一个左括号为止,计算结果放到nums- 数字 : 从当前位置开始继续往后取,将整一个连续数字整体取出,加入
nums + - * / ^ %: 需要将操作放入ops中。在放入之前先把栈内可以算的都算掉(只有「栈内运算符」比「当前运算符」优先级高/同等,才进行运算),使用现有的nums和ops进行计算,直到没有操作或者遇到左括号,计算结果放到nums
我们可以通过 🌰 来理解 只有「栈内运算符」比「当前运算符」优先级高/同等,才进行运算 是什么意思:
因为我们是从前往后做的,假设我们当前已经扫描到 2 + 1 了(此时栈内的操作为 + )。
- 如果后面出现的
+ 2或者- 1的话,满足「栈内运算符」比「当前运算符」优先级高/同等,可以将2 + 1算掉,把结果放到nums中; - 如果后面出现的是
* 2或者/ 1的话,不满足「栈内运算符」比「当前运算符」优先级高/同等,这时候不能计算2 + 1。
更为详细的讲解可以看这篇题解 :使用「双栈」解决「究极表达式计算」问题
代码:
class Solution {
public int clumsy(int n) {
Deque<Integer> nums = new ArrayDeque<>();
Deque<Character> ops = new ArrayDeque<>();
// 维护运算符优先级
Map<Character, Integer> map = new HashMap<>(){{
put('*', 2);
put('/', 2);
put('+', 1);
put('-', 1);
}};
char[] cs = new char[]{'*', '/', '+', '-'};
for (int i = n, j = 0; i > 0; i--, j++) {
char op = cs[j % 4];
nums.addLast(i);
// 如果「当前运算符优先级」不高于「栈顶运算符优先级」,说明栈内的可以算
while (!ops.isEmpty() && map.get(ops.peekLast()) >= map.get(op)) {
calc(nums, ops);
}
if (i != 1) ops.add(op);
}
// 如果栈内还有元素没有算完,继续算
while (!ops.isEmpty()) calc(nums, ops);
return nums.peekLast();
}
void calc(Deque<Integer> nums, Deque<Character> ops) {
int b = nums.pollLast(), a = nums.pollLast();
int op = ops.pollLast();
int ans = 0;
if (op == '+') ans = a + b;
else if (op == '-') ans = a - b;
else if (op == '*') ans = a * b;
else if (op == '/') ans = a / b;
nums.addLast(ans);
}
}
- 时间复杂度:
- 空间复杂度:
数学解法(打表技巧分析)
这次在讲【证明】之前,顺便给大家讲讲找规律的题目该怎么做。
由于是按照特定顺序替换运算符,因此应该是有一些特性可以被我们利用的。
通常我们需要先实现一个可打表的算法(例如上述的解法一,这是为什么掌握「通用表达式」解法具有重要意义),将连续数字的答案打印输出,来找找规律:
Solution solution = new Solution();
for (int i = 1; i <= 10000; i++) {
int res = solution.clumsy(i);
System.out.println(i + " : " + res);
}
似乎 与 答案比较接近,我们考虑将两者的差值输出:
Solution solution = new Solution();
for (int i = 1; i <= 10000; i++) {
int res = solution.clumsy(i);
System.out.println(i + " : " + res + " : " + (res - i));
}
咦,好像发现了什么不得了的东西。似乎每四个数,差值都是 [1, 2, 2, -1]
再修改我们的打表逻辑,来验证一下(只输出与我们猜想不一样的数字):
Solution solution = new Solution();
int[] diff = new int[]{1,2,2,-1};
for (int i = 1; i <= 10000; i++) {
int res = solution.clumsy(i);
int t = res - i;
if (t != diff[i % 4]) {
System.out.println(i + " : " + res);
}
}
只有前四个数字被输出,其他数字都是符合我们的猜想规律的。
到这里我们已经知道代码怎么写可以 AC 了,十分简单。
代码:
class Solution {
public int clumsy(int n) {
int[] special = new int[]{1,2,6,7};
int[] diff = new int[]{1,2,2,-1};
if (n <= 4) return special[(n - 1) % 4];
return n + diff[n % 4];
}
}
- 时间复杂度:
- 空间复杂度:
证明
讲完我们的【实战技巧】之后,再讲讲如何证明。
上述的做法比较适合于笔试或者比赛,但是面试,通常还需要证明做法为什么是正确的。
我们不失一般性的分析某个 n,当然这个 n 必须是大于 4,不属于我们的特判值。
然后对 n 进行讨论(根据我们的打表猜想去证明规律是否可推广):
-
n % 4 == 0: ,即diff = 1 -
n % 4 == 1: ,即diff = 2 -
n % 4 == 2: ,即diff = 2 -
n % 4 == 3: ,即diff = -1
上述的表达式展开过程属于小学数学内容,省略号部分的项式的和为 0,因此你只需要关注我写出来的那部分。
至此,我们证明了我们的打表猜想具有「可推广」的特性。
甚至我们应该学到:证明可以是基于猜想去证明,而不必从零开始进行推导。
224. 基本计算器
给你一个字符串表达式 s ,请你实现一个基本计算器来计算并返回它的值。
示例 1:
输入:s = "1 + 1"
输出:2
示例 2:
输入:s = " 2-1 + 2 "
输出:3
示例 3:
输入:s = "(1+(4+5+2)-3)+(6+8)"
输出:23
提示:
s由数字、'+'、'-'、'('、')'、和' '组成s表示一个有效的表达式
双栈解法
我们可以使用两个栈 nums 和 ops 。
nums: 存放所有的数字ops:存放所有的数字以外的操作,+/-也看做是一种操作
然后从前往后做,对遍历到的字符做分情况讨论:
- 空格 : 跳过
(: 直接加入ops中,等待与之匹配的)): 使用现有的nums和ops进行计算,直到遇到左边最近的一个左括号为止,计算结果放到nums- 数字 : 从当前位置开始继续往后取,将整一个连续数字整体取出,加入
nums +/-: 需要将操作放入ops中。在放入之前先把栈内可以算的都算掉,使用现有的nums和ops进行计算,直到没有操作或者遇到左括号,计算结果放到nums
一些细节:
- 由于第一个数可能是负数,为了减少边界判断。一个小技巧是先往
nums添加一个 0 - 为防止 () 内出现的首个字符为运算符,将所有的空格去掉,并将
(-替换为(0-,(+替换为(0+(当然也可以不进行这样的预处理,将这个处理逻辑放到循环里去做)
Java 代码:
class Solution {
public int calculate(String s) {
// 存放所有的数字
Deque<Integer> nums = new ArrayDeque<>();
// 为了防止第一个数为负数,先往 nums 加个 0
nums.addLast(0);
// 将所有的空格去掉
s = s.replaceAll(" ", "");
// 存放所有的操作,包括 +/-
Deque<Character> ops = new ArrayDeque<>();
int n = s.length();
char[] cs = s.toCharArray();
for (int i = 0; i < n; i++) {
char c = cs[i];
if (c == '(') {
ops.addLast(c);
} else if (c == ')') {
// 计算到最近一个左括号为止
while (!ops.isEmpty()) {
char op = ops.peekLast();
if (op != '(') {
calc(nums, ops);
} else {
ops.pollLast();
break;
}
}
} else {
if (isNum(c)) {
int u = 0, j = i;
// 将从 i 位置开始后面的连续数字整体取出,加入 nums
while (j < n && isNum(cs[j])) u = u * 10 + (int)(cs[j++] - '0');
nums.addLast(u);
i = j - 1;
} else {
if (i > 0 && (cs[i - 1] == '(' || cs[i - 1] == '+' || cs[i - 1] == '-')) {
nums.addLast(0);
}
// 有一个新操作要入栈时,先把栈内可以算的都算了
while (!ops.isEmpty() && ops.peekLast() != '(') calc(nums, ops);
ops.addLast(c);
}
}
}
while (!ops.isEmpty()) calc(nums, ops);
return nums.peekLast();
}
void calc(Deque<Integer> nums, Deque<Character> ops) {
if (nums.isEmpty() || nums.size() < 2) return;
if (ops.isEmpty()) return;
int b = nums.pollLast(), a = nums.pollLast();
char op = ops.pollLast();
nums.addLast(op == '+' ? a + b : a - b);
}
boolean isNum(char c) {
return Character.isDigit(c);
}
}
C++ 代码:
class Solution {
public:
void replace(string& s){
int pos = s.find(" ");
while (pos != -1) {
s.replace(pos, 1, "");
pos = s.find(" ");
}
}
int calculate(string s) {
// 存放所有的数字
stack<int> nums;
// 为了防止第一个数为负数,先往 nums 加个 0
nums.push(0);
// 将所有的空格去掉
replace(s);
// 存放所有的操作,包括 +/-
stack<char> ops;
int n = s.size();
for(int i = 0; i < n; i++) {
char c = s[i];
if(c == '(')
ops.push(c);
else if(c == ')') {
// 计算到最近一个左括号为止
while(!ops.empty()) {
char op = ops.top();
if(op != '(')
calc(nums, ops);
else {
ops.pop();
break;
}
}
}
else {
if(isdigit(c)) {
int cur_num = 0, j = i;
// 将从 i 位置开始后面的连续数字整体取出,加入 nums
while(j <n && isdigit(s[j]))
cur_num = cur_num*10 + (s[j++] - '0');
// 注意上面的计算一定要有括号,否则有可能会溢出
nums.push(cur_num);
i = j - 1;
}
else {
if (i > 0 && (s[i - 1] == '(' || s[i - 1] == '+' || s[i - 1] == '-')) {
nums.push(0);
}
// 有一个新操作要入栈时,先把栈内可以算的都算了
while(!ops.empty() && ops.top() != '(')
calc(nums, ops);
ops.push(c);
}
}
}
while(!ops.empty())
calc(nums, ops);
return nums.top();
}
void calc(stack<int> &nums, stack<char> &ops) {
if(nums.size() < 2 || ops.empty())
return;
int b = nums.top(); nums.pop();
int a = nums.top(); nums.pop();
char op = ops.top(); ops.pop();
nums.push(op == '+' ? a+b : a-b);
}
};
- 时间复杂度:
- 空间复杂度:
进阶
- 如果在此基础上,再考虑
*和/,需要增加什么考虑?如何维护运算符的优先级? - 在 的基础上,如果考虑支持自定义符号,例如
a / func(a, b) * (c + d),需要做出什么调整?
补充
- 对应进阶 1 的补充。
一个支持 + - * / ^ % 的「计算器」,基本逻辑是一样的,使用字典维护一个符号优先级:
class Solution {
Map<Character, Integer> map = new HashMap<>(){{
put('-', 1);
put('+', 1);
put('*', 2);
put('/', 2);
put('%', 2);
put('^', 3);
}};
public int calculate(String s) {
s = s.replaceAll(" ", "");
char[] cs = s.toCharArray();
int n = s.length();
Deque<Integer> nums = new ArrayDeque<>();
nums.addLast(0);
Deque<Character> ops = new ArrayDeque<>();
for (int i = 0; i < n; i++) {
char c = cs[i];
if (c == '(') {
ops.addLast(c);
} else if (c == ')') {
while (!ops.isEmpty()) {
if (ops.peekLast() != '(') {
calc(nums, ops);
} else {
ops.pollLast();
break;
}
}
} else {
if (isNumber(c)) {
int u = 0;
int j = i;
while (j < n && isNumber(cs[j])) u = u * 10 + (cs[j++] - '0');
nums.addLast(u);
i = j - 1;
} else {
if (i > 0 && (cs[i - 1] == '(' || cs[i - 1] == '+' || cs[i - 1] == '-')) {
nums.addLast(0);
}
while (!ops.isEmpty() && ops.peekLast() != '(') {
char prev = ops.peekLast();
if (map.get(prev) >= map.get(c)) {
calc(nums, ops);
} else {
break;
}
}
ops.addLast(c);
}
}
}
while (!ops.isEmpty() && ops.peekLast() != '(') calc(nums, ops);
return nums.peekLast();
}
void calc(Deque<Integer> nums, Deque<Character> ops) {
if (nums.isEmpty() || nums.size() < 2) return;
if (ops.isEmpty()) return;
int b = nums.pollLast(), a = nums.pollLast();
char op = ops.pollLast();
int ans = 0;
if (op == '+') {
ans = a + b;
} else if (op == '-') {
ans = a - b;
} else if (op == '*') {
ans = a * b;
} else if (op == '/') {
ans = a / b;
} else if (op == '^') {
ans = (int)Math.pow(a, b);
} else if (op == '%') {
ans = a % b;
}
nums.addLast(ans);
}
boolean isNumber(char c) {
return Character.isDigit(c);
}
}
- 关于进阶 ,其实和进阶 一样,重点在于维护优先级。但还有一些编码细节:
对于非单个字符的运算符(例如 函数名function),可以在处理前先将所有非单字符的运算符进行替换(将 function 替换为 @# 等)
然后对特殊运算符做特判,确保遍历过程中识别到特殊运算符之后,往后整体读入(如 function(a,b) -> @(a, b),@(a, b) 作为整体处理)
227. 基本计算器 II
给你一个字符串表达式 s ,请你实现一个基本计算器来计算并返回它的值。
整数除法仅保留整数部分。
示例 1:
输入:s = "3+2*2"
输出:7
示例 2:
输入:s = " 3/2 "
输出:1
示例 3:
输入:s = " 3+5 / 2 "
输出:5
提示:
s由整数和算符 ('+','-','*','/') 组成,中间由一些空格隔开s表示一个 有效表达式- 表达式中的所有整数都是非负整数,且在范围 内
- 题目数据保证答案是一个
32-bit整数
双栈
如果你有看这篇 224. 基本计算器 的话,今天这道题就是道练习题。
帮你巩固 双栈解决「通用表达式」问题的通用解法 。
事实上,我提供这套解决方案不仅仅能解决只有 + - ( )的 [224. 基本计算器] 或者 + - * / [227. 基本计算器 II(本题)] 的表达式问题,还能解决 + - * / ^ % ( ) 的完全表达式问题。
甚至支持自定义运算符,只要在运算优先级上进行维护即可。
对于「表达式计算」这一类问题,你都可以使用这套思路进行解决。我十分建议你加强理解这套处理逻辑。
对于「任何表达式」而言,我们都使用两个栈 nums 和 ops:
nums: 存放所有的数字ops:存放所有的数字以外的操作
然后从前往后做,对遍历到的字符做分情况讨论:
- 空格 : 跳过
(: 直接加入ops中,等待与之匹配的)): 使用现有的nums和ops进行计算,直到遇到左边最近的一个左括号为止,计算结果放到nums- 数字 : 从当前位置开始继续往后取,将整一个连续数字整体取出,加入
nums + - * / ^ %: 需要将操作放入ops中。在放入之前先把栈内可以算的都算掉(只有「栈内运算符」比「当前运算符」优先级高/同等,才进行运算),使用现有的nums和ops进行计算,直到没有操作或者遇到左括号,计算结果放到nums
我们可以通过 🌰 来理解 只有「栈内运算符」比「当前运算符」优先级高/同等,才进行运算 是什么意思:
因为我们是从前往后做的,假设我们当前已经扫描到 2 + 1 了(此时栈内的操作为 + )。
- 如果后面出现的
+ 2或者- 1的话,满足「栈内运算符」比「当前运算符」优先级高/同等,可以将2 + 1算掉,把结果放到nums中; - 如果后面出现的是
* 2或者/ 1的话,不满足「栈内运算符」比「当前运算符」优先级高/同等,这时候不能计算2 + 1。
一些细节:
- 由于第一个数可能是负数,为了减少边界判断。一个小技巧是先往
nums添加一个 0 - 为防止 () 内出现的首个字符为运算符,将所有的空格去掉,并将
(-替换为(0-,(+替换为(0+(当然也可以不进行这样的预处理,将这个处理逻辑放到循环里去做) - 从理论上分析,
nums最好存放的是long,而不是int。因为可能存在大数 + 大数 + 大数 + … - 大数 - 大数的表达式导致中间结果溢出,最终答案不溢出的情况
代码:
class Solution {
// 使用 map 维护一个运算符优先级
// 这里的优先级划分按照「数学」进行划分即可
Map<Character, Integer> map = new HashMap<>(){{
put('-', 1);
put('+', 1);
put('*', 2);
put('/', 2);
put('%', 2);
put('^', 3);
}};
public int calculate(String s) {
// 将所有的空格去掉
s = s.replaceAll(" ", "");
char[] cs = s.toCharArray();
int n = s.length();
// 存放所有的数字
Deque<Integer> nums = new ArrayDeque<>();
// 为了防止第一个数为负数,先往 nums 加个 0
nums.addLast(0);
// 存放所有「非数字以外」的操作
Deque<Character> ops = new ArrayDeque<>();
for (int i = 0; i < n; i++) {
char c = cs[i];
if (c == '(') {
ops.addLast(c);
} else if (c == ')') {
// 计算到最近一个左括号为止
while (!ops.isEmpty()) {
if (ops.peekLast() != '(') {
calc(nums, ops);
} else {
ops.pollLast();
break;
}
}
} else {
if (isNumber(c)) {
int u = 0;
int j = i;
// 将从 i 位置开始后面的连续数字整体取出,加入 nums
while (j < n && isNumber(cs[j])) u = u * 10 + (cs[j++] - '0');
nums.addLast(u);
i = j - 1;
} else {
if (i > 0 && (cs[i - 1] == '(' || cs[i - 1] == '+' || cs[i - 1] == '-')) {
nums.addLast(0);
}
// 有一个新操作要入栈时,先把栈内可以算的都算了
// 只有满足「栈内运算符」比「当前运算符」优先级高/同等,才进行运算
while (!ops.isEmpty() && ops.peekLast() != '(') {
char prev = ops.peekLast();
if (map.get(prev) >= map.get(c)) {
calc(nums, ops);
} else {
break;
}
}
ops.addLast(c);
}
}
}
// 将剩余的计算完
while (!ops.isEmpty()) calc(nums, ops);
return nums.peekLast();
}
void calc(Deque<Integer> nums, Deque<Character> ops) {
if (nums.isEmpty() || nums.size() < 2) return;
if (ops.isEmpty()) return;
int b = nums.pollLast(), a = nums.pollLast();
char op = ops.pollLast();
int ans = 0;
if (op == '+') ans = a + b;
else if (op == '-') ans = a - b;
else if (op == '*') ans = a * b;
else if (op == '/') ans = a / b;
else if (op == '^') ans = (int)Math.pow(a, b);
else if (op == '%') ans = a % b;
nums.addLast(ans);
}
boolean isNumber(char c) {
return Character.isDigit(c);
}
}
- 时间复杂度:
- 空间复杂度:
总结
综上,使用三叶提供的这套「双栈通用解决方案」,可以解决所有的「表达式计算」问题。
因为这套「表达式计算」处理逻辑,本质上模拟了人脑的处理逻辑:根据下一位的运算符优先级决定当前运算符是否可以马上计算。