题解 | #牛牛计算器# java
牛牛计算器
https://www.nowcoder.com/practice/192ac31d5e054abcaa10b72d9b01cace
import java.util.*;
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param s string字符串
* @return int整型
*/
public int calculate (String s) {
// write code here
Stack<Integer> nums = new Stack<>();
int num = 0;
char op = '+';
for (int i = 0; i < s.length(); i++) {
char c = s.charAt(i);
if (Character.isDigit(c)) {
num = num * 10 + (c - '0');
} else if (c == '(') {
int j = i, cnt = 0;
for (; i < s.length(); i++) {
if (s.charAt(i) == '(') cnt++;
if (s.charAt(i) == ')') cnt--;
if (cnt == 0) {
break;
}
}
num = calculate(s.substring(j + 1, i));
}
if (!Character.isWhitespace(c) || i == s.length() - 1) {
if (op == '+') {
nums.push(num);
} else if (op == '-') {
nums.push(-num);
} else if (op == '*') {
int temp = nums.pop() * num;
nums.push(temp);
}
num = 0;
op = c;
}
}
int res = 0;
while (!nums.empty()) {
res += nums.pop();
}
return res;
}
}
编程语言是Java。
该题考察的知识点包括:
- 使用栈(Stack)数据结构进行数学表达式的计算
- 字符串的遍历和判断
- 字符转数字的处理
- 子字符串的提取
代码的文字解释如下:
- 遍历字符串
s中的每一个字符c:如果 c 是数字,则将其转换为整数并与 num 相加。如果 c 是左括号 (,则找到与之匹配的右括号 ) 的位置,在这个子串上递归调用 calculate 方法,并将结果赋值给 num。如果 c 是运算符或者是字符串 s 的最后一个字符,则根据当前的运算符 op 将 num 入栈:如果 op 是加号 +,则将 num 入栈。如果 op 是减号 -,则将 -num 入栈。如果 op 是乘号 *,则弹出栈顶元素与 num 相乘,并将结果入栈。将 num 重置为0,将 op 更新为当前字符 c。 - 循环结束后,栈中剩余的元素分别是各个部分的计算结果,将其依次弹出并累加到变量
res中。
