224. 基本计算器【力扣(LeetCode) 】

理论最高的吻

于 2024-09-04 21:24:34 发布

阅读量387

点赞数 12

分类专栏：编程练习【LeetCode（力扣）】文章标签： leetcode 算法职场和发展 C++ 栈逆波兰表达式

本文链接：https://blog.csdn.net/yyh520025/article/details/141816465

版权

编程练习【LeetCode（力扣）】专栏收录该内容

57 篇文章 0 订阅

订阅专栏

一、题目描述

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

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

二、测试用例

示例 1：

输入：s = "1 + 1"
输出：2

示例 2：

输入：s = " 2-1 + 2 "
输出：3

示例 3：

输入：s = "(1+(4+5+2)-3)+(6+8)"
输出：23

提示：

1 <= s.length <= 3 * 105
s 由数字、'+'、'-'、'('、')'、和 ' ' 组成
s 表示一个有效的表达式
'+' 不能用作一元运算(例如， "+1" 和 "+(2 + 3)" 无效)
'-' 可以用作一元运算(即 "-1" 和 "-(2 + 3)" 是有效的)
输入中不存在两个连续的操作符
每个数字和运行的计算将适合于一个有符号的 32位 整数

三、解题思路

【注：官方的方法没有怎么复杂，我这个更容易理解一些】

基本思路：
先除去算术表达式中的空格，并将对应元素分割，为了区别负号和减号，负号使用 n 表示；将分割好的算术表达式转为逆波兰表达式；最后用栈完成计算。
具体思路：
- 格式化算术表达式：遍历字符串，
  - 遇到空格就跳过
  - 遇到数字就拼接到字符串
  - 遇到符号先判断数字拼接的字符串是否为空，不为空，则先存入数字拼接字符串，在存入符号；为空则直接存入符号。
- 转为逆波兰表达式：遍历格式化的算术表达式，
  - 遇到数字则存入数字栈中
  - 遇到符号则进行判断：
    - 符号栈为空，符号为左括号，符号优先级小于符号栈顶元素——符号入栈
    - 符号为右括号——将符号栈中元素依次弹入到数字栈中，直到遇到左括号为止
    - 符号优先级大于等于符号栈顶元素——弹出符号栈顶元素，存入到数字栈中；再将符号入栈
  - 遍历结束后将符号栈中剩余元素弹入到数字栈中
- 用栈计算逆波兰表达式：详情看文章：150. 逆波兰表达式求值【力扣(LeetCode) 】【注意负号的处理即可】

四、参考代码

时间复杂度： $\Omicron(n)$
空间复杂度： $\Omicron(n)$

class Solution {
public:
    map<string, int> op_set = {
        {"n", 1}, {"+", 1}, {"-", 1}, {"(", 2}, {")", 2}};

    vector<string> trim(string s) {  // 将算术表达分割并删去空格
        vector<string> ans;
        string t = "";

        for (auto c : s) {
            if (isdigit(c)) {
                t += c;
            } else if (c != ' ') {
                if (t != "") {
                    ans.push_back(t);
                    t = "";
                }
                if (c == '-' && (ans.empty() || ans.back() == "(")) { // 负号的情况
                    c = 'n';
                }
                ans.push_back(string(1, c));
            }
        }
        if (t != "") {
            ans.push_back(t);
        }

        return ans;
    }

    vector<string> s2nbl(vector<string> s) { // 将分割后的字符串转为逆波兰表达式
        vector<string> num, op;
        string t = "";

        for (auto& c : s) {
            if (op_set.count(c) == 1) {
                if (op.empty() || c == "(" || op_set[c] < op_set[op.back()]) {
                    op.push_back(c);
                } else {
                    if (c == ")") {
                        while (op.back() != "(") {
                            num.push_back(op.back());
                            op.pop_back();
                        }
                        op.pop_back();
                    } else {
                        num.push_back(op.back());
                        op.pop_back();
                        op.push_back(c);
                    }
                }
            } else {
                num.push_back(c);
            }
        }
        while (!op.empty()) {
            num.push_back(op.back());
            op.pop_back();
        }

        return num;
    }

    int calculate(string s) {
        int a, b;
        vector<int> num;
        vector<string> strs = s2nbl(trim(s));

        for (auto& str : strs) {
            if (op_set.count(str) == 1) {
                a = num.back();
                num.pop_back();
                if (str != "n") {
                    b = num.back();
                    num.pop_back();
                }
                if (str == "+")
                    num.push_back(a + b);
                else if (str == "-")
                    num.push_back(b - a);
                else
                    num.push_back(-a);
            } else {
                istringstream is(str);
                is >> a;
                num.push_back(a);
            }
        }

        return num.back();
    }
};