LeetCode 每日一题227. 基本计算器 II

227. 基本计算器 II

给你一个字符串表达式 s ，请你实现一个基本计算器来计算并返回它的值。

整数除法仅保留整数部分。

示例 1：

输入：s = "3+2*2"
输出：7

示例 2：

输入：s = " 3/2 "
输出：1

示例 3：

输入：s = " 3+5 / 2 "
输出：5

提示：

• 1 <= s.length <= 3 * 105
• s 由整数和算符 (’+’, ‘-’, ‘*’, ‘/’) 组成，中间由一些空格隔开
• s 表示一个 有效表达式
• 表达式中的所有整数都是非负整数，且在范围 [0, 231 - 1] 内
• 题目数据保证答案是一个 32-bit 整数

方法一：

解题思路

• 只有加减乘除，没有括号，也没有负数。遍历数组，先计算遇到的乘法/除法，剩下的数字按照加法处理即可。

参考代码

public int calculate(String s) {
    Deque<Integer> nums = new LinkedList<>();
    char sign = '+';
    int num = 0;
    int n = s.length();
    for (int i = 0; i < n; i++) {
        char c = s.charAt(i);
        if (Character.isDigit(c)) {
            num = num * 10 + (c - '0');
        }
        if ((c != ' ' && !Character.isDigit(c)) || i == n - 1) {
            if (sign == '+') {
                nums.push(num);
            } else if (sign == '-') {
                nums.push(-num);
            } else if (sign == '*') {
                nums.push(nums.pop() * num);
            } else if (sign == '/') {
                nums.push(nums.pop() / num);
            }
            sign = c;
            num = 0;
        }
    }
    return nums.stream().mapToInt(e -> e).sum();
}

执行结果

• 有一说一，stream 的效率是真的低，不过爽就完事了～