LeetCode——计算表达式

最新推荐文章于 2022-02-06 17:46:51 发布

csdn_ggboy

最新推荐文章于 2022-02-06 17:46:51 发布

阅读量365

点赞数

分类专栏： LeetCode 文章标签： LeetCode 基本计算器双栈表达式计算运算符优先级

本文链接：https://blog.csdn.net/csdn_ggboy/article/details/121580573

版权

LeetCode 专栏收录该内容

13 篇文章 0 订阅

订阅专栏

224. 基本计算器 - 力扣（LeetCode）

【进阶补充】双栈解决通用「表达式计算」问题

class Solution {
public:
    void replace(string &s)
    {
        int pos = s.find(" ");
        while(pos != -1)
        {
            s.replace(pos, 1, "");
            pos = s.find(" ");
        }
    }
    int calculate(string s) {
        stack<int> nums;
        nums.push(0);
        replace(s);
        stack<char> ops;
        int n = s.size();
        for(int i = 0; i < n; i ++ )
        {
            char c = s[i];
            if(c == '(') ops.push(c);
            else if(c == ')')
            {
                while(!ops.empty())
                {
                    char op = ops.top();
                    if(op == '(')
                    {
                        ops.pop();
                        break;
                    }
                    else calc(nums, ops);
                }
            }
            else
            {
                if(isdigit(c))
                {
                    int cur_num = 0;
                    int j = i;
                    //找数
                    while(j < n && isdigit(s[j]))
                        cur_num = cur_num * 10 + (s[j ++] - '0');
                    nums.push(cur_num);
                    i = j - 1;
                }  
                else
                {
                    if(i > 0 && s[i - 1] == '(')
                        nums.push(0);
                    while(!ops.empty() && ops.top() != '(')
                        calc(nums, ops);
                    ops.push(c);
                }
            }
        }
        while(!ops. empty())
            calc(nums, ops);
        return nums.top();
    }
    void calc(stack<int> &nums, stack<char> &ops)
    {
        if(nums.size() < 2 || ops.empty()) return;

        int b = nums.top(); nums.pop();
        int a = nums.top(); nums.pop();
        char op = ops.top(); ops.pop();
        nums.push(op == '+' ? a + b : a - b);
    }
};

227. 基本计算器 II 题解 - 力扣（LeetCode）

【宫水三叶】使用「双栈」解决「究极表达式计算」问题 - 基本计算器 II - 力扣（LeetCode）

class Solution {
public:
    void replace(string &s)
    {
        int k = s.find(" ");
        while(k != -1)
        {
            s.replace(k, 1, "");
            k = s.find(" ");
        }
    }
    unordered_map<char, int> ops_pri = {
        {'+', 1},
        {'-', 1},
        {'*', 2},
        {'/', 2},
        {'%', 2},
        {'^', 3}
    };//定义优先级
    stack<int> ops;
    stack<long long> nums;
    int calculate(string s) {
        nums.push(0);//防止第一个数为负
        replace(s);//删除所有空格
        int n = s.size();
        for(int i = 0; i < n; i ++ )
        {
            char c = s[i];
            if(c == '(')//直接放进去
                ops.push(c);
            else if(c == ')')//计算到(
            {
                while(!ops.empty())
                {
                    char op = ops.top();
                    if(op == '(')
                    {
                        ops.pop();
                        break;
                    }
                    else calc();
                }
            }
            else
            {
                if(isdigit(c))//读取数字
                {
                    int cur_num = 0;
                    int j = i;
                    while(j < n && isdigit(s[j]))
                        cur_num = cur_num * 10 + (s[j ++] - '0');
                    nums.push(cur_num);
                    i = j - 1;
                }
                else
                {
                    if(i > 0 && s[i - 1] == '(')
                        nums.push(0);
                    while(!ops.empty() && ops.top() != '(' && ops_pri[ops.top()] >= ops_pri[c])//站内运算符优先级高
                        calc();
                    ops.push(c);
                }
            }
        }
        while(!ops.empty())
            calc();
        return nums.top();
    }
    void calc()
    {
        long long b = nums.top(); nums.pop();
        long long a = nums.top(); nums.pop();
        char op = ops.top(); ops.pop();
        long long res;
        switch(op)
        {
            case '+': res = a + b; break;
            case '-': res = a - b; break;
            case '*': res = a * b; break;
            case '/': res = a / b; break;
            case '%': res = a % b; break;
            case '^': res = a ^ b; break;
        }
        nums.push(res);
    }
};