通过模拟的方式来处理。
设置A队列用来存放题目数据,B栈用来放1到n这n个数,S栈作为中转栈。
通过分析,不难发现
当A栈不为空并且
【S栈为空或者S的栈顶元素值小于A队列第一个元素的值】并且
S栈不满时:
我们让B栈弹出一个元素压入S栈中。
当上述循环跳出时有以下4种可能:
(1)A队列为空,说明YES
(2)S的栈顶元素值等于A队列首元素值,则S弹出栈顶,A队首元素出列
(3)S的栈顶元素值大于A队列首元素值,说明NO
(4)S种的元素已经达到最大容量,说明NO。
按照上述的分析,我们写出下面的代码:
#include<cstdio>
#include<queue>
#include<stack>
using namespace std;
int main()
{
int capacity,n,k,buff;
bool flag;
scanf("%d%d%d",&capacity,&n,&k);
for(int i=0; i<k; i++)
{
stack<int> B,S;
queue<int> A;
for(int j=n; j>=1; j--)
{
scanf("%d",&buff);
A.push(buff);
B.push(j);
}
while(1)
{
while(A.size()!=0&&(S.size()==0||S.top()<A.front())&&S.size()<capacity){
S.push(B.top());B.pop();
}
if(A.size()==0){
flag=true;break;
}
else if(S.top()==A.front()){
S.pop();A.pop();
}
else if(S.top()>A.front()){
flag=false;break;
}
else if(S.size()==capacity){
flag=false;break;
}
}
if(flag)printf("YES\n");
else printf("NO\n");
}
return 0;
}