题目
给你一个字符串数组 tokens
,表示一个根据 逆波兰表示法 表示的算术表达式。
请你计算该表达式。返回一个表示表达式值的整数。
注意:
- 有效的算符为
'+'
、'-'
、'*'
和'/'
。 - 每个操作数(运算对象)都可以是一个整数或者另一个表达式。
- 两个整数之间的除法总是 向零截断 。
- 表达式中不含除零运算。
- 输入是一个根据逆波兰表示法表示的算术表达式。
- 答案及所有中间计算结果可以用 32 位 整数表示。
示例 1:
输入:tokens = ["2","1","+","3","*"] 输出:9 解释:该算式转化为常见的中缀算术表达式为:((2 + 1) * 3) = 9
示例 2:
输入:tokens = ["4","13","5","/","+"] 输出:6 解释:该算式转化为常见的中缀算术表达式为:(4 + (13 / 5)) = 6
示例 3:
输入: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
C++代码
#include <iostream>
#include <string>
#include <vector>
#include <stack>
using namespace std;
/*
* 逆波兰表达式求值问题
* 如果字符串数组里的元素为数字,将字符串转化为数值放入栈中
* 如果为4种符号,从栈顶取出num1,num2进行求值
*/
bool isNumber(string& token) {
return !(token == "+" || token == "-" || token == "*" || token == "/");
}
int evalRPN(vector<string>& tokens) {
stack<int> stk;
int n = tokens.size();
for (int i = 0; i < n; ++i) {
string& token = tokens[i];
if (isNumber(token)) {
stk.push(atoi(token.c_str()));
}
else {
int num2 = stk.top();
stk.pop();
int num1 = stk.top();
stk.pop();
switch (token[0]) {
case '+':
stk.push(num1 + num2);
break;
case '-':
stk.push(num1 - num2);
break;
case '*':
stk.push(num1 * num2);
break;
case '/':
stk.push(num1 / num2);
break;
}
}
}
return stk.top();
}
int main() {
vector<string> tokens = { "2","1","+","3","*" };
int ans = evalRPN(tokens);
cout << ans << endl;
return 0;
}
分析
逆波兰表达式求值问题,如果字符串数组里的元素为数字,将字符串转化为数值放入栈中;如果为4 种符号,从栈顶取出num1,num2进行求值。
问题
c_str() 函数是C++中 string 类中的函数,将 string 转换为 const char* 。
atoi() 函数把字符串转换成整型数。