227. 基本计算器 II

image.png

题解

由于乘除优先于加减计算,因此不妨考虑先进行所有乘除运算,并将这些乘除运算后的整数值放回原表达式的相应位置,则随后整个表达式的值,就等于一系列整数加减后的值。
基于此,我们可以用一个栈,保存这些(进行乘除运算后的)整数的值。
对于加减号后的数字,将其直接压入栈中;对于乘除号后的数字,可以直接与栈顶元素计算,并替换栈顶元素为计算后的结果。具体来说,遍历字符串 s,并用变量 preSign 记录每个数字之前的运算符,对于第一个数字,其之前的运算符视为加号。
每次遍历到数字末尾时,根据 preSign 来决定计算方式:

  • 加号:将数字压入栈;
  • 减号:将数字的相反数压入栈;
  • 乘除号:计算数字与栈顶元素,并将栈顶元素替换为计算结果。

代码实现中,若读到一个运算符,或者遍历到字符串末尾,即认为是遍历到了数字末尾。处理完该数字后,更新 preSign 为当前遍历的字符。
遍历完字符串 s 后,将栈中元素累加,即为该字符串表达式的值。

执行用时:11 ms, 在所有 Java 提交中击败了80.14%的用户 内存消耗:38.5 MB, 在所有 Java 提交中击败了81.38%的用户

  1. class Solution {
  2. public int calculate(String s) {
  3. int preSign = '+';
  4. int num = 0;
  5. int n = s.length();
  6. Stack<Integer> stack = new Stack<>();
  7. for (int i = 0; i < n; i++) {
  8. char c = s.charAt(i);
  9. if (Character.isDigit(c)) {
  10. num = (num * 10) + (c - '0');
  11. }
  12. if (!Character.isDigit(c) && c != ' ' || (i == n - 1)) {
  13. if (preSign == '+') {
  14. stack.push(num);
  15. } else if (preSign == '-') {
  16. stack.push(-num);
  17. } else if (preSign == '*') {
  18. stack.push(stack.pop() * num);
  19. } else if (preSign == '/') {
  20. stack.push(stack.pop() / num);
  21. } else {
  22. throw new IllegalArgumentException("未知的运算符:" + c);
  23. }
  24. preSign = c;
  25. num = 0;
  26. }
  27. }
  28. int res = 0;
  29. while (!stack.isEmpty()) {
  30. res += stack.pop();
  31. }
  32. return res;
  33. }
  34. }