201903-2 二十四点

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;
}

题目链接