【LeetCode】224. 基本计算器 Basic Calculator II(C++)

目录

• • 题目描述
  • 题目大意
  • 栈
  • 复杂度分析

题目来源：https://leetcode-cn.com/problems/basic-calculator-ii/

题目描述

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

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

示例 1：

输入：s = "3+22"
输出：7
示例 2：

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

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

提示：

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

题目大意

• 比昨天的每日一题要简单许多，只需要动态维护数据栈和操作符栈就可以了，然后注意当数据栈中只剩一个数且字符栈op的只剩下’-'时，此时符号需要取反

栈

typedef long long ll;
class Solution {
public:
    int calculate(string s) {
        stack<char> op;
        stack<ll> nums;
        int i = 0, len = s.size();
        while(i < len){
            if(s[i] == ' ') ++i;
            if(s[i] == '+' || s[i] == '-' || s[i] == '*' || s[i] == '/'){
                op.push(s[i]);
                ++i;
            }
            if(s[i] >= '0' && s[i] <= '9'){
                ll num = 0;
                while(s[i] >= '0' && s[i] <= '9'){
                    num =  num * 10 + s[i] - '0';
                    ++i;
                }
                if(!op.empty() && op.top() == '-'){
                   num *= -1;
                   op.pop();
                   op.push('+');
                }
                nums.push(num);
                if(!op.empty() && (op.top() == '*' || op.top() == '/')){
                    ReallyCalculate(op.top(), nums);
                    op.pop();
                }
            }
        }
        // 当数据栈中只剩一个操作数时出来
        while(nums.size() != 1){
            ReallyCalculate(op.top(), nums);
            op.pop();
        }
        int ans = nums.top();
        if(nums.size() == 1 && !op.empty() && op.top() == '-')
            ans *= -1;
        return ans;
    }
    void ReallyCalculate(char op, stack<ll>& nums){
        ll secondNum = nums.top();
        nums.pop();
        ll firstNum = nums.top();
        nums.pop();
        switch(op){
            case '*':
                nums.push(firstNum * secondNum);
                break;
            case '/':
                nums.push(firstNum / secondNum);
                break;
            case '+':
                nums.push(firstNum + secondNum);
                break;
            case '-':
                nums.push(firstNum - secondNum);
                break;
        }
    }
};

复杂度分析

• 时间复杂度：O(n)。n为数组的长度
• 空间复杂度：O(n)。n为数组的长度

标签：nums,Calculator,C++,II,&&,push,firstNum,top,op
来源： https://blog.csdn.net/lr_shadow/article/details/114652716