227. 基本计算器 II
给你一个字符串表达式 s ,请你实现一个基本计算器来计算并返回它的值。
整数除法仅保留整数部分。
示例 1:
输入:s = "3+2*2"
输出:7
示例 2:
输入:s = " 3/2 "
输出:1
示例 3:
输入:s = " 3+5 / 2 "
输出:5
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/basic-calculator-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
代码
class Solution {
public:
int calculate(string s) {
typedef long long LL;
stack<LL>nums,rnums;//准备两个数据栈和两个符号栈
stack<char>sign,rsign;
char record[s.size()+10];//使用一个数组来过滤掉空格
LL temp=0;
int count=0;
for(int i=0;i<s.size();i++)//如果字符i不是空格,则加入到数组中
if(s[i]!=' ')record[count++]=s[i];
for(int i=0;i<count;i++)
{
if(record[i]>='0'&&record[i]<='9'){ while(record[i]>='0'&&record[i]<='9'){//当是数字是,通过循环得到这个数的实际的值
temp=temp*10+(record[i]-'0');
i++;
}
nums.push(temp);//将该数推入栈
temp=0;
i--;//i当前为数字的下一位,多滑动了一位,需向后退
continue;
}
if(record[i]=='*'||record[i]=='/'){
//如果当前符号是*或/,则优先级最高,可以直接计算
int a=nums.top();//a为栈顶元素
nums.pop();
int b=0;//b为符号后面跟着的数
char sign1=record[i];
while(record[i+1]>='0'&&record[i+1]<='9'){
b=b*10+(record[i+1]-'0');
i++;
}
if(sign1=='*')a=a*b;
else a=a/b;
nums.push(a);
continue;
}
if(record[i]=='+'||record[i]=='-')sign.push(record[i]);//如果是+或-,优先级较低,推入符号栈
}
while(!nums.empty()){//将符号栈和数字栈调转,顺序完成计算
rnums.push(nums.top());
nums.pop();
}
while(!sign.empty()){
rsign.push(sign.top());
sign.pop();
}
while(!rsign.empty()){
int a=rnums.top();
rnums.pop();
int b=rnums.top();
rnums.pop();
if(rsign.top()=='+')a=a+b;
else a=a-b;
rnums.push(a);
rsign.pop();
}
return rnums.top();
}
};