LeetCode 227——基本计算器II（双栈）

一、题目介绍

实现一个基本的计算器来计算一个简单的字符串表达式的值。

字符串表达式仅包含非负整数，+， - ，*，/ 四种运算符和空格  。 整数除法仅保留整数部分。

示例 1:

输入: "3+2*2"
输出: 7
示例 2:

输入: " 3/2 "
输出: 1
示例 3:

输入: " 3+5 / 2 "
输出: 5
说明：

你可以假设所给定的表达式都是有效的。
请不要使用内置的库函数 eval。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/basic-calculator-ii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、解题思路

（1）建立两个栈，一个存放运算数，一个存放运算符。

（2）利用栈后进先出的特性，通过比较当前运算符与栈顶运算符的优先级，执行计算过程。

       ① 当前运算符的优先级高于栈顶运算符：将当前运算符压栈。

       ② 当前运算符的优先级不大于栈顶运算符：从存放运算数的栈中取出左右操作数，从存放运算符的栈中取出操作符，计算                之后将结果压入存放运算符的栈中，再将当前运算符压栈。

（3）按照上面的原则，实现基本计算器，注意空字符。

三、解题代码

class Solution {
private:
    map<char, int> mp;
    stack<int> nums;
    stack<char> symbol;
public:
    void helper(int x, bool flag)
    {
        while(!symbol.empty())
        {
            if(flag == true)
            {
                int temp = x - mp[symbol.top()];
                if(temp >= 2) //后面运算符的优先级大于前面的
                {
                    break;
                }
            }
            //后面的运算符优先级不大于前面的优先级
            int rn = nums.top();
            nums.pop();
            int ln = nums.top();
            nums.pop();
            switch (mp[symbol.top()])
            {
                case 1:
                    nums.push(ln + rn);
                    break;
                case 2:
                    nums.push(ln - rn);
                    break;
                case 4:
                    nums.push(ln * rn);
                    break;
                case 5:
                    nums.push(ln / rn);
                    break;
            }
            symbol.pop();
        }
    }
    int calculate(string s) {
        mp['+'] = 1;
        mp['-'] = 2;
        mp['*'] = 4;
        mp['/'] = 5;
        int n = s.size();
        int val = 0;
        for(int i = 0; i < n; ++i)
        {
            if(s[i] >= '0' && s[i] <= '9')
            {
                val *= 10;
                val += s[i] - '0';
            }          
			else if(mp[s[i]])
            {
                nums.push(val);
                val = 0;
                if(symbol.empty())   //第一个计算符
                    symbol.push(s[i]);
                else    //考虑优先级
                {
                    helper(mp[s[i]], true);
                    symbol.push(s[i]);
                }
            }
        }
        nums.push(val); //考虑最后一个运算数
        helper(0, false);
        return nums.top();
    }
};

四、解题结果