一、题目描述
传送门:150. 逆波兰表达式求值
描述:
根据 逆波兰表示法,求表达式的值。
有效的算符包括 +、-、*、/ 。每个运算对象可以是整数,也可以是另一个逆波兰表达式。
注意 两个整数之间的除法只保留整数部分。
可以保证给定的逆波兰表达式总是有效的。换句话说,表达式总会得出有效数值且不存在除数为 0 的情况
示例:
输入:tokens = ["2","1","+","3","*"]
输出:9
解释:该算式转化为常见的中缀算术表达式为:((2 + 1) * 3) = 9
输入:tokens = ["10","6","9","3","+","-11","*","/","*","17","+","5","+"]
输出:22
解释:该算式转化为常见的中缀算术表达式为:
((10 * (6 / ((9 + 3) * -11))) + 17) + 5
= ((10 * (6 / (12 * -11))) + 17) + 5
= ((10 * (6 / -132)) + 17) + 5
= ((10 * 0) + 17) + 5
= (0 + 17) + 5
= 17 + 5
= 22
二、思路
该题目为后缀表达式求值,可以使用栈进行实现,
当输入为数字时候,压入栈
输入为运算符时候,则聪栈中弹出两个运算数进行运算,结果再次压入栈中,最后留在栈中的元素即为结果
注意的是,由于栈是先进后出,所以弹出的第一个数其实是第二个操作数
示例:
例如:tokens = [“2”,“1”,“+”],在遇到“+”之前,栈中元素是2,1
遇到“+”,则先后弹出1,2
计算结果:1 + 2 = 3
三、代码
C++代码如下:
class Solution {
public:
int evalRPN(vector<string>& tokens) {
stack<int> st;
for(string s : tokens) {
if(s == "+" || s == "-" || s == "*" || s == "/") {
int b = st.top(); // 先出来的是第二个数
st.pop();
int a = st.top(); // 第一个数
st.pop();
if(s == "+") st.push(a + b);
else if(s == "-") st.push(a - b);
else if(s == "*") st.push(a * b);
else st.push(a / b);
}
else st.push(stoi(s)); // string -> int
}
return st.top();
}
};