1051 Pop Sequence (25分)(栈混洗)

通过模拟的方式来处理。

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

猜你喜欢

转载自blog.csdn.net/yiwaite/article/details/105414301