PTA出栈序列的合法性

给定一个最大容量为 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

思路：数组 a 存放待比较的出栈序列， pos 表示数组 a 当前要出栈的数的下标

st 是容量为 m 的栈， 初始时 1 入栈。 num 顺序取自 1~n ，初值为 1

关键思想就是不断出栈和不断入栈，当既不能出栈也不能入栈，退出循环

循环后根据是否栈空判断 YES/NO

AC代码： 

#include<bits/stdc++.h>
using namespace std;

int main()
{
	int m,n,k;
	cin>>m>>n>>k;
	int a[1001];
	while(k--)
	{
		for(int i=1;i<=n;i++) cin>>a[i];
		stack<int>st;
		st.push(1);
		int num=2,pos=1;
		int flag=1;
		while(flag)
		{
			flag=0;
			while(!st.empty()&&pos<=n&&st.top()==a[pos])
			{
				st.pop();
				pos++;
				flag=1;
			}
			while(st.size()<m&&num<=n&&num<=a[pos])
			{
				st.push(num);
				num++;
				flag=1;
			}
		}
		if(st.empty()) cout<<"YES\n";
		else cout<<"NO\n";
	}
    return 0;
}

 下面是一道类似的栈的题目：

接发球 

有n个篮球，它们的编号是1~n。我们按照编号顺序给你发球。你每接到一个球，你能做两个事情：要么把手上的这个球发出去，要么把球放入到球框（如果还放得下的话）。如果你现在手上没有球，你可以选择接下一个球或者从球筐里取出一个球发出去。
现按照编号指定一个发球顺序，请你帮忙完成任务。
注意：我们只提供一个球筐，这个球框最多只能放m个球，框开口向上，每层只能放一个球。所以取球时只能从上往下顺序取。
好了，请你告诉我能完成任务吗，能的话，输出YES，否则，输出NO。

输入格式:

第一行有三个整数，n, m, k（0<=n<=100, 0<m<=100），分别代表篮球数量，球框容量，以及有多少次询问。
接下来有k行，每行是n个数，代表我们指定的一个编号排列，这个排列就是我们要求的发球顺序。

输出格式:

输出有k行，每行输出对应排列是否能完成。能的话，输出YES，否则，输出NO.

输入样例:

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

输出样例:

YES
NO

结果说明：
1、对于3 2 1 4 6 5，可以接1放框里，接2放框里，接到的3发出去，再把框里的2和1发出去，然后接4，发出去，再接6放框里，接5发出去，再把框里的6发出去。
2、对于1 6 5 4 3 2，可以接1发出去，然后三次接球放框，再接到的5已经放不下了，所以无法完成任务。

AC代码：

#include<bits/stdc++.h>
using namespace std;
int n,m,k;
bool ju(int a[])
{
	stack<int>s;
	int now=1;
	int flag=1;
	int i=0;
	while(i<n&&flag)
	{
		if(now==a[i])
		{
			now++;
			i++;
		}
		else if(now<a[i])//此时要入栈
		{
			while(now<a[i]&&s.size()<m)
			s.push(now++);
			if(now<a[i]) flag=0;//说明栈满了
			else i++,now++;
		}
		else //此时要出栈
		{
			if(s.top()!=a[i]) flag=0;
			else
			{
				s.pop();
				i++;
			}
		}
	}
	if(!flag) return false;
	while(!s.empty())//还得比较出栈序列是否正确
	{
		if(s.top()!=a[i]) return false;
		s.pop();
		i++;
	}
	return true;
}
int main()
{
	cin>>n>>m>>k;
	int a[101];
	for(int i=0;i<k;i++)
	{
		for(int j=0;j<n;j++) cin>>a[j];
		if(ju(a)) cout<<"YES"<<endl;
		else cout<<"NO"<<endl;
	}
}