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

一、题目描述

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

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

二、测试用例

示例 1:

cpp 复制代码
输入:s = "1 + 1"
输出:2

示例 2:

cpp 复制代码
输入:s = " 2-1 + 2 "
输出:3

示例 3:

cpp 复制代码
输入:s = "(1+(4+5+2)-3)+(6+8)"
输出:23

提示:

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

三、解题思路

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

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

四、参考代码

时间复杂度: O ( n ) \Omicron(n) O(n)

空间复杂度: O ( n ) \Omicron(n) O(n)

cpp 复制代码
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();
    }
};
相关推荐
林泽毅2 分钟前
SwanLab x EasyR1:多模态LLM强化学习后训练组合拳,让模型进化更高效
算法·llm·强化学习
小林熬夜学编程4 分钟前
【高并发内存池】第八弹---脱离new的定长内存池与多线程malloc测试
c语言·开发语言·数据结构·c++·算法·哈希算法
余华余华5 分钟前
2024年蓝桥杯Java B组省赛真题超详解析-分布式队列
java·职场和发展·蓝桥杯
刚入门的大一新生11 分钟前
归并排序延伸-非递归版本
算法·排序算法
独好紫罗兰15 分钟前
洛谷题单3-P1980 [NOIP 2013 普及组] 计数问题-python-流程图重构
开发语言·python·算法
独好紫罗兰20 分钟前
洛谷题单3-P1009 [NOIP 1998 普及组] 阶乘之和-python-流程图重构
开发语言·python·算法
曦月逸霜32 分钟前
蓝桥杯高频考点——高精度(含C++源码)
c++·算法·蓝桥杯
ゞ 正在缓冲99%…41 分钟前
leetcode152.乘积最大子数组
数据结构·算法·leetcode
敲上瘾1 小时前
高并发内存池(二):Central Cache的实现
linux·服务器·c++·缓存·哈希算法
独行soc1 小时前
2025年渗透测试面试题总结-某腾某讯-技术安全实习生升级(题目+回答)
java·python·安全·web安全·面试·职场和发展·红蓝攻防