传送门
Description
小Z设计了一种新的数据结构“新壳栈”。首先,它和传统的栈一样支持压入、弹出操作。此外,其栈顶的前c个元素是它的壳,支持翻转操作。其中,c>2是一个固定的正整数,表示壳的厚度。小Z还希望,每次操作,无论是压入、弹出还是翻转,都仅用与c无关的常数时间完成。聪明的你能帮助她编程实现“新壳栈”吗?
程序期望的实现效果如以下两表所示。其中,输入的第一行是正整数c,之后每行输入都是一条指令。另外,如遇弹出操作时栈为空,或翻转操作时栈中元素不足c个,应当输出相应的错误信息。
指令 涵义
1[空格]e 在栈顶压入元素e
2 弹出(并输出)栈顶元素
3 翻转栈顶的前c个元素
0 退出
Input
第一行输入c,之后每行输入都是一条指令,输入以0结束。
Output
对于每个弹出操作,输出栈顶元素,如果栈为空输出“Error: the stack is empty!”。对于每个翻转操作,如果栈中元素不足c个,输出”Error: less than c elements in the stack!“,注意这里的c在输出时以具体输入的c的值代替。
Sample Input
3
1 1
1 2
1 3
1 4
3
1 5
3
2
2
2
3
2
2
2
0
Sample Output
3
2
5
Error: less than 3 elements in the stack!
4
1
Error: the stack is empty!
Data Constraint
c<=50000
操作数<=1000000
Analysis
一眼题,直接上暴力,正解是双向队列+栈???
有点小猥琐,反转时用了c++自带函数
Realization
#include<bits/stdc++.h>
using namespace std;
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
#define fin(x) freopen(""#x".in","r",stdin)
#define fout(x) freopen(""#x".out","w",stdout)
#define open(x) fin(x);fout(x)
#define ll long long
#define inf INT_MAX
int c,p,t[1000001];
int main(){
scanf("%d",&c);
while(1){
scanf("%d",&p);
if(p==0) return 0;
if(p==1){
scanf("%d",&p);
t[++t[0]]=p;
continue;
}
if(p==2){
if(t[0]==0) printf("Error: the stack is empty!\n");
else printf("%d\n",t[t[0]--]);
continue;
}
if(p==3){
if(t[0]<c) printf("Error: less than %d elements in the stack!\n",c);
else reverse(t+t[0]-c+1,t+t[0]+1);
}
}
}