2021-04-29

 

L2-1 出栈序列的合法性

题目来源：2020年团体程序设计天梯赛-模拟赛

给定一个最大容量为 M 的堆栈，将 N 个数字按 1, 2, 3, ..., N 的顺序入栈，允许按任何顺序出栈，则哪些数字序列是不可能得到的？例如给定 M=5、N=7，则我们有可能得到{ 1, 2, 3, 4, 5, 6, 7 }，但不可能得到{ 3, 2, 1, 7, 5, 6, 4 }。

输入格式：

输入第一行给出 3 个不超过 1000 的正整数：M（堆栈最大容量）、N（入栈元素个数）、K（待检查的出栈序列个数）。最后 K 行，每行给出 N 个数字的出栈序列。所有同行数字以空格间隔。

输出格式：

对每一行出栈序列，如果其的确是有可能得到的合法序列，就在一行中输出YES，否则输出NO。

输入样例：

5 7 5
1 2 3 4 5 6 7
3 2 1 7 5 6 4
7 6 5 4 3 2 1
5 6 4 3 7 2 1
1 7 6 5 4 3 2

输出样例：

YES
NO
NO
YES
NO

思路：1. 栈是一种先入先出的线性表，只能在顶部进行插入或删除操作。

           2. 本题中我们设置两个数组a[N]和b[N]，其中a用来保存入栈顺序的元素(从0到n)，b用来保存出栈顺序的元素(从0到n)，在设置一个栈sta（初始为空）和两个下标 i (入栈顺序数组的下标，初始为0)和 j (出栈顺序数组的下标，初始为0)；

           3.从下标0开始遍历b数组的每一个元素b[j]，每一次判断 b[j] 是否与栈sta顶元素相等，若相等则将栈顶元素pop出来并且判断下一个b中的元素（即j++），若不相等或者栈为空则将a中的元素a[i] push进去（栈未满时）；不断循环直到判断完b中的每一个元素后（即j<n不成立时）结束，此时该出栈顺序合法；若循环未遍历到b中的每一个元素而是以 栈已满push不进去 或 栈未满但已无元素可入栈（即 i<n不成立）而结束，此时该出栈顺序不合法；

参考代码：

#include <iostream>    //vxgzh：xtsn
using namespace std;
#include <stack>
#define N 1005

int n,m;    //m:栈的容量，n:元素个数 
int a[N];   //入栈顺序（从0到N） 
int b[N];	//出栈顺序（从0到N）

bool abc()    //栈序列的合法性判断，合法返回true，否则返回false 
{
	int i=0,j=0;    //i：入栈数组下标， j：入栈数组下标
	stack <int> sta;	
	while(j<n)
	{	
		if(!sta.empty()&&sta.top()==b[j])   //判断是否等于栈顶元素 
		{
			sta.pop();j++;		     //出栈，出栈指针后移 
		}
		else if(i<n&&sta.size()<m)    //如果栈未满，按入栈顺序将元素入栈 
		{
			sta.push(a[i++]);
		}
		else       //栈已满或者栈未满但无元素可入栈 
			return false;		
	}
	return true;	
}

int main()
{
	int k,i,j;
	cin>>m>>n>>k;
	for(i=0;i<n;i++)  //入栈顺序数组a
		a[i]=i+1;
	for(i=0;i<k;i++)
	{
		for(j=0;j<n;j++)  //出栈顺序数组b
			cin>>b[j];
		if(abc())
			cout<<"YES"<<endl;	
		else
			cout<<"NO"<<endl;
	}
	
	return 0;
 }