class Solution {
public:
long long compute(const char& c, long long num1, long long num2)
{
if (c == '+')
return num2 + num1;
else if (c == '-')
return num2 - num1;
else if (c == '*')
return num2 * num1;
else
return num2 / num1;
}
int evalRPN(vector<string>& tokens) {
stack<long long> st;
for (auto& s : tokens)
{
if (!isdigit(s[0]) && s.size() == 1)//字符串首字符不是数字且长度为1 说明是运算符
{
int num1 = st.top();
st.pop();
int num2 = st.top();
st.pop();
st.push(compute(s[0], num1, num2));
}
else//数字字符串
{
int num = 0, flag = 1;
size_t i = 0;
if (s[i] == '-')
{
flag = -1;
i++;
}
//将数字字符串转换为整型推入栈中
for (; i < s.size(); i++)
{
num = num * 10 + (s[i] - '0');
}
num *= flag;
st.push(num);
}
}
return st.top();
}
//方法2 利用function包装器和map建立符号和操作方法的映射
int evalRPN(vector<string>& tokens) {
stack<int> st;
//建立符号和操作方法的映射
map<string, function<int(int, int)>> m = {
{ "+", [](int a, int b){ return a + b; } },
{ "-", [](int a, int b){ return a - b; } },
{ "*", [](int a, int b){ return a * b; } },
{ "/", [](int a, int b){ return a / b; } }
};
for (auto& str : tokens)
{
if (m.find(str) != m.end())
{
int a = st.top();
st.pop();
int b = st.top();
st.pop();
st.push(m[str](b, a));
}
else
{
st.push(stoi(str));
}
}
return st.top();
}
};
