洛谷 P2186 小Z的栈函数
题目
题目描述
小Z最近发现了一个神奇的机器，这个机器的所有操作都是通过维护一个栈来完成的，它支持如下11个操作：
NUM X：栈顶放入X。
POP：抛弃栈顶元素。
INV：将栈顶元素取出，然后放入它的相反数。
DUP：再放入一个和栈顶元素相同的数。
SWP：交换栈顶的两个元素。
ADD：取出栈顶的两个元素，两元素相加，所得结果放入栈内。
SUB：取出栈顶的两个元素，第二个元素减去第一个元素，所得结果放入栈内。
MUL：取出栈顶的两个元素，两元素相乘，所得结果放入栈内。
DIV：取出栈顶的两个元素，第二个元素整除以第一个元素，所得结果放入栈内。
MOD：取出栈顶的两个元素，第二个元素取模以第一个元素，所得结果放入栈内。
END：结束这个程序。
然后，小Z用上面的11种操作写了一个一元函数f(x)。x就是放入栈里面第一个初始元素。然后经过这个函数的一系列操作，当函数结束的时候，正常情况下，栈里面会有唯一的一个元素。剩下的这个元素就作为函数f(x)的返回值。
小Z有N个询问，询问每个值x经过上述函数所映射出的f(x)是多少。但是这个由于机器太老了，跑起东西来太慢了，小Z又是一个急性子，所以请你们写一个程序，来帮助小Z计算他查询的f(x)。
输入输出格式
输入格式：
输入若干行，仅包含上述11个操作，用来描述函数f(x)的操作，函数的结束保证以END结尾；
接下来一个整数N；
下面N行每行一个数字ai，代表栈里面的初始元素。
输入数据不保证合法！！！
输出格式：
如果最后栈内不是一个元素，输出“ERROR”；
还有，由于这台机器太破了，所以如果运算过程中有数字的绝对值大于1000000000机器也输出“ERROR”；
如果输入数据不合法，导致中途退出，输出“ERROR”；
否则输出对应的f(x)。
输入输出样例
输入样例#1：
NUM 600000000
ADD
END
3
0
600000000
1输出样例#1：
600000000
ERROR
600000001说明
【提示】
仔细考虑不合法的情况，避免不必要的RE和WA。
【数据规模】
函数操作步数<=2000
询问数<=2000
题解
一道很水的模拟题，就是代码很烦…
代码
#include <cstdio>
#include <string>
#include <iostream>
#include <stack>
#define INF 1000000000
#define maxn 2005
using namespace std;
int tot,n,flag;
int a[maxn];
stack<int> stk;
struct des{
int num;
string s;
} q[maxn];
void end()
{
printf("ERROR\n");
}
bool check(int x)
{
if (x>INF||x<-INF) return true;
else return false;
}
void num(int x)
{
if (check(q[x].num)) flag=1;
else stk.push(q[x].num);
}
void inv()
{
int x=stk.top();
if (check(-x)) flag=1;
else {
stk.pop();
stk.push(-x);
}
}
void dup()
{
int x=stk.top();
stk.push(x);
}
void swp()
{
int x,y;
x=stk.top();stk.pop();
y=stk.top();stk.pop();
if (check(x)||check(y)) flag=1;
else {
stk.push(x);stk.push(y);
}
}
void add()
{
int x,y;
x=stk.top();stk.pop();
y=stk.top();stk.pop();
if (check(x+y)) flag=1;
else stk.push(x+y);
}
void sub(){
int x,y;
x=stk.top();stk.pop();
y=stk.top();stk.pop();
if (check(y-x)) flag=1;
else stk.push(y-x);
}
void mul()
{
int x,y;
x=stk.top();stk.pop();
y=stk.top();stk.pop();
if (check(x*y)) flag=1;
else stk.push(x*y);
}
void div()
{
int x,y;
x=stk.top();stk.pop();
y=stk.top();stk.pop();
if (!x||check(y/x)) flag=1;
else stk.push(y/x);
}
void mod()
{
int x,y;
x=stk.top();stk.pop();
y=stk.top();stk.pop();
if (!x&&check(y%x)) flag=1;
else stk.push(y%x);
}
int main(){
tot=0;
while(true){
cin>>q[tot].s;
if (q[tot].s=="NUM") scanf("%d",&q[tot].num);
if (q[tot].s=="END") break;
else tot++;
}
scanf("%d",&n);
for (int i=0;i<n;i++) scanf("%d",&a[i]);
for (int i=0;i<n;i++){
while (stk.size()!=0) stk.pop();
stk.push(a[i]);
flag=0;
for (int j=0;j<tot;j++){
if (check(a[i])){
flag=1;break;
}
if (q[j].s=="NUM") num(j);
else if (q[j].s=="POP") stk.pop();
else if (q[j].s=="INV") inv();
else if (q[j].s=="DUP") dup();
else if (q[j].s=="SWP") swp();
else if (q[j].s=="ADD") add();
else if (q[j].s=="SUB") sub();
else if (q[j].s=="MUL") mul();
else if (q[j].s=="DIV") div();
else if (q[j].s=="MOD") mod();
if (flag==1) break;
}
if (flag==1){
end();continue;
}
if (stk.size()!=1) end();
else printf("%d\n",stk.top());
}
return 0;
}