LeetCode每日一题题解：227. 基本计算器 II-题解-python && C++源代码

227. 基本计算器 II

难度中等541收藏分享切换为英文接收动态反馈

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

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

你可以假设给定的表达式总是有效的。所有中间结果将在 [-231, 231 - 1] 的范围内。

注意：不允许使用任何将字符串作为数学表达式计算的内置函数，比如 eval() 。

示例 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

题解：

单调栈的思想，我们遇到数字就赋给新变量num，然后继续判断，如果遇见符号，赋给plus
nums就是遇见一个新符号之前的数子，plus则是遇见新符号之前的一个符号，
我们在遇见一个新符号的时候，判断之前的符号是什么符号，然后对num进行相应的操作
stack的栈顶的值为plus前一个数。

Python代码：

class Solution:
    def calculate(self, s):
        stack = []  #设置栈用来存储数组
        plus = '+'  #设首个符号为+号，方便首次将第一个数压入栈
        num = 0  #设初始值为0
        for i , j in enumerate(s):
            if j.isdigit():  #判断是否为数字，如果是位置，则将值赋给num
               num = num*10 + int(j)        #这句的原因主要在于有的测试用例是两个数中间没有符号的，比如“42”这样的
            if i == len(s)-1 or j in '+-*/':   #如果到最后一个数了，或者j为符号，需要进行以下比较
                if plus == '+':            #如果为 ‘+’ 直接压入栈
                    stack.append(num)      
                elif plus == '-':      #如果为‘-’ 则将num变为-num再压入栈
                    stack.append(-num)
                elif plus == '*':  #如果为‘*’，则弹出栈顶的值与num相乘
                    stack.append(stack.pop()*num)
                elif plus == '/':   #如果为‘/’,则弹出栈顶的值，判断top是否小于0，如果小于0，则进行/  否则//
                    top = stack.pop()
                    if top<0:
                        stack.append(int(top/num))
                    else:
                        stack.append(top//num)
                plus = j  #这次的符号赋给plus
                num = 0  #num重新置0
        return sum(stack)  #对stack求和

C++代：

class Solution {
public:
    int calculate(string s) {
        vector<int> stacks;
        char plus = '+'; //#设置栈用来存储数组
        int num = 0;   // #设首个符号为+号，方便首次将第一个数压入栈
        int len = s.length();
        for (int i = 0; i<len; i++){
            if (isdigit(s[i])){    //#判断是否为数字，如果是位置，则将值赋给num
                num = num*10 + int(s[i] - '0');    // #这句的原因主要在于有的测试用例是两个数中间没有符号的，比如“42”这样的
            }
            if (!isdigit(s[i]) && s[i] != ' ' || i == len - 1){  // #如果到最后一个数了，或者j为符号，需要进行以下比较
                if (plus == '+'){  // 如果为 ‘+’ 直接压入栈
                    stacks.push_back(num);
                }
                else if (plus == '-'){ //  如果为‘-’ 则将num变为-num再压入栈
                    stacks.push_back(-num);
                }
                else if (plus == '*'){  // 如果为‘*’，则弹出栈顶的值与num相乘
                    stacks.back() *= num;
                }
                else if (plus == '/'){ // 如果为‘/’,则弹出栈顶的值，进行除法
                    stacks.back() /= num;
                }
            plus = s[i];
            num = 0;

            }
        }
        return accumulate(stacks.begin() , stacks.end() , 0);
    }
};