201903-2 二十四点
题意描述
本题要求我们实现一个简单的计算器。
输入输出格式
输入一个n
,表示要计算的字符串的个数,接下来是n
个字符串
数据规模
n <= 1e2
算法设计
本题的核心类似于中缀表达式转为后缀表达式,因为计算机处理后缀表达式相当简单,只要维护一个栈即可。这里关键是如何进行转化。运算符有优先级,因此我们维护unordered_map<char,int>um
来表示优先级。设置一个运算符栈op
,一个运算数栈num
。遍历表达式,如果是数字,直接压入运算数栈,如果是运算符,这里要分类讨论,如果现运算符必栈顶运算符优先级高,显然,该运算符应当入栈,应当先于栈顶运算符进行运算。但如果优先级小于等于栈顶运算符,那栈应该从栈顶开始进行运算,并将结果重新压入栈中,直至栈顶运算符优先级低于现运算符,或是栈为空。之后,再将现运算符压入栈中。如果遍历完字符串后,运算符栈不为空,则直接从栈顶开始进行计算即可。最后操作数栈只有一个结果,即为运算答案。
c++11代码
#include <bits/stdc++.h>
using namespace std;
using gg = long long;
gg compute(gg a, gg b,char c){
if(c == '+') return a+b;
if(c == '-') return a-b;
if(c == 'x') return a*b;
if(c == '/') return a/b;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
gg n;
cin>>n;
string s;
unordered_map<char,gg>um{{'+',0},{'-',0},{'x',1},{'/',1}}; //优先级
while(n--){
cin>>s;
stack<char>op; // 操作符栈
stack<gg>num; // 操作数栈
for(auto i : s){
if(isdigit(i)) num.push(i - '0');
else {
while(not op.empty() and um[op.top()] >= um[i]){ // 进行计算
gg a = num.top();
num.pop();
gg b = num.top();
num.pop();
num.push(compute(b,a,op.top())); // 注意操作数顺序,小心除法
op.pop();
}
op.push(i);
}
}
while(not op.empty()){
gg a = num.top();
num.pop();
gg b = num.top();
num.pop();
num.push(compute(b,a,op.top()));
op.pop();
}
cout<< (num.top() == 24 ? "Yes\n" : "No\n");
}
return 0;
}