PTA 7-6 出栈序列的合法性

题目描述

给定一个最大容量为 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，…，n的顺序；第二种情况就是前k个元素一直入栈，然后开始出栈，那么出栈序列的顺序就是k，k-1，……，1；第三种情况就是前面这两种情况的结合。





退出循环的条件是i>n或者A满了，但是判断是否退出循环的先决条件是i不等于B和A的pop不等于B。

源代码

#include<stdio.h>
 
int m,n,k;
int stack[1007];       //栈 
int top=0;             //栈顶下标 
int index1=1,index2=1;     //index1来表示1-n的数字 ，index2来表示数组b的下标 
int b[1007];           //存放出栈序列 
 
int main()
{
    scanf("%d%d%d",&m,&n,&k);
    while(k--)
    {
    	int flag=1; 
    	index1=1;index2=1;
    	top=0; 
    	int i;
    	for(i=1;i<=n;i++)
    	  scanf("%d",&b[i]);
    	while(1)
    	{
    		//判断给出的出栈序列如果和1到n增长顺序相同，那就是入栈一个元素，然后立即出栈。 
    		if(index1==b[index2])
    		{
    			index1++;
    			index2++;
			}
			//判断如果栈中有元素，然后栈顶元素也与此时的出栈序列元素相同，那么就继续出栈来判断 
			else if(top!=0&&stack[top-1]==b[index2])
			{
				top--;
				index2++; 
			}
			else
			{
				if(index1>n)break;  //已经判断结束，就跳出循环 
				stack[top]=index1;  //元素入栈 
				top++;
				index1++;
				if(top>=m) //
				{
					flag=0;
					break;
				}
			}
		}
		if(flag==0||top!=0)printf("NO\n");
		else printf("YES\n");
	}
    
	return 0;
}