【leetcode】逆波兰表达式求值
引言:
在力扣网刷题时找了一道中等难度的题,其实在我看来应该划归为简单栏的,下面我们就一起来看一下这个题的思路和代码。
思路:
逆波兰式:也就是后缀表达式(将运算符写在操作数之后) 计算逆波兰式需要用到一个栈(用于存入计算用到的数值)。
计算逆波兰式依据以下规则: 遍历用户输入的逆波兰式。 如果遍历到的元素是一个数值,那么把该元素入栈等待后续计算;
如果遍历到的元素是一个符号,那么出栈栈顶的两个数值元素用该符号进行计算。
(注意:由于栈的后进先出特性,先出栈的是第二个操作数,后出栈的是第一个操作数-在减法、除法运算时有不同)
再把结果计算结果入栈,等待下一个计算。当遍历完整个输入的逆波兰式后,当前栈顶数值就是总的计算结果。
代码如下:
class Solution {
public:
int evalRPN(vector<string>& tokens) {
stack<int> s;
for(size_t i=0;i<tokens.size();++i)
{
if(tokens[i].size()==1 &&
( tokens[i][0]=='+'
|| tokens[i][0]=='-'
|| tokens[i][0]=='*'
|| tokens[i][0]=='/'))
{
int right=s.top();
s.pop();
int left=s.top();
s.pop();
if(tokens[i][0]=='+')
s.push(left+right);
else if(tokens[i][0]=='-')
s.push(left-right);
else if(tokens[i][0]=='*')
s.push(left*right);
else if(tokens[i][0]=='/')
s.push(left/right);
else
assert(false);
}
else
{
s.push(atoi(tokens[i].c_str()));
}
}
return s.top();
}
};