一、题目介绍
实现一个基本的计算器来计算一个简单的字符串表达式的值。
字符串表达式仅包含非负整数,+, - ,*,/ 四种运算符和空格 。 整数除法仅保留整数部分。
示例 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();
}
};