AcWing 3273. 二十四点(表达式求值)基础课板子题目csp16(2)
原题链接
中等
作者:
YAX_AC
,
2024-11-26 18:05:27
,
所有人可见
,
阅读 5
#include <iostream>
#include <algorithm>
#include <cstring>
#include <stack>
#include <unordered_map>
using namespace std;
stack<char> op; //操作符栈
stack<int> num; //数字栈
void eval()
{
auto b = num.top(); num.pop();
auto a = num.top(); num.pop();
auto c = op.top(); op.pop();
int x;
if(c == '+') x = a + b;
else if(c == '-') x = a - b;
else if(c == 'x') x = a * b;
else x = a / b;
num.push(x);
}
int main()
{
int T;
cin >> T;
while (T -- )
{
string s;
cin >> s;
op = stack<char>();
num = stack<int>();
unordered_map<char, int> pr{{'+', 1}, {'-', 1}, {'x', 2}, {'/', 2}}; //定义运算符优先级
for(int i = 0; i < s.size(); i ++ )
{
if(isdigit(s[i])) //是数,压入数字栈
{
int j = i, x = 0;
while(isdigit(s[j]) && j < s.size())
x = x * 10 + s[j ++ ] - '0';
num.push(x);
i = j - 1;
}
else if(s[i] == '(') //是左括号,压入操作符栈
{
op.push(s[i]);
}
else if(s[i] == ')') //是右括号,计算前面的数直到遇到左括号为止
{
while(op.top() != '(') eval();
op.pop(); //弹出左括号
}
else //是操作符
{
// 栈顶操作符优先级大于等于当前操作符优先级,则要先计算
while(op.size() && op.top() != '(' && pr[op.top()] >= pr[s[i]])
eval();
op.push(s[i]);
}
}
while(op.size()) eval();
if(num.top() == 24) puts("Yes");
else puts("No");
}
return 0;
}