原题链接
这几天都在刷stack模块,感觉还行。
这道题与括号匹配差不多,都是属于栈的经典例题。
遍历整个tokens,判断当前的元素ch是否为操作符:
如果不是,就调用stoi函数将ch转为数字类型,并压栈;
否则两次出栈,分别得到两次的栈顶元素,进行相应的就算,最后返回栈顶。
class Solution {
public:
int evalRPN(vector<string>& tokens) {
vector<int> num;
for(auto ch :tokens){
if(ch!="/"&&ch!="*"&&ch!="+"&&ch!="-"){
num.push_back(stoi(ch));
}
else {
int num1 = num.back();num.pop_back();
int num2 = num.back(); num.pop_back();
int sum=0;
if(ch=="+")sum=num1+num2;
else if(ch=="-")sum = num2-num1;
else if(ch=="*")sum = num1*num2;
else sum = num2/num1;
num.push_back(sum);
}
}
return num.back();
}
};