PAT A 1051. Pop Sequence (25)

题目

Given a stack which can keep M numbers at most.  Push N numbers in the order of 1, 2, 3, ..., N and pop randomly.  You are supposed to tell if a given sequence of numbers is a possible pop sequence of the stack.  For example, if M is 5 and N is 7, we can obtain 1, 2, 3, 4, 5, 6, 7 from the stack, but not 3, 2, 1, 7, 5, 6, 4.

Input Specification:

Each input file contains one test case.  For each case, the first line contains 3 numbers (all no more than 1000): M (the maximum capacity of the stack), N (the length of push sequence), and K (the number of pop sequences to be checked).  Then K lines follow, each contains a pop sequence of N numbers.  All the numbers in a line are separated by a space.

Output Specification:

For each pop sequence, print in one line "YES" if it is indeed a possible pop sequence of the stack, or "NO" if not.

Sample Input:

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

Sample Output:

YES
NO
NO
YES
NO

 

判断一个序列是否可以成为1~n的数压入后的弹出情况。

记录已经出现过的最大的数stack_max（即为已经压入到的数），堆栈中的元素数量，上一次弹出的数。

 

1、出现一个超过已经出现过的最大的数stack_max的数mm，意味着要压入stack_max+1~mm，判断容量问题，刷新数据，继续。

2、出现一个小于上次弹出的数，弹出，刷新数据，继续。

3、其他，错误，跳出。

 

这里的空间要求相当宽裕，实际可以用一个堆栈来模拟，判断下堆栈大小、将要压入的数、栈顶的数是否与序列中的下一个数匹配即可。

 

代码：

#include <iostream>
using namespace std;

const int MAX=0x37777777;
bool Stack_judge(int m,int n,int data[1001]);	//判断是否符合要求

int main()
{
	int m,n,k;	//输入数据
	int data[1001];		//单次的随机排列数据
	cin>>m>>n>>k;

	int i,j;
	for(i=0;i<k;i++)	//输入并判断
	{
		for(j=0;j<n;j++)
			scanf("%d",&data[j]);
		if(Stack_judge(m,n,data))
			printf("YES\n");
		else
			printf("NO\n");
	}

	return 0;
}

bool Stack_judge(int m,int n,int data[1001])
{
	int stack_max=0;	//当前堆栈中出现过的最大数字
	int stack_num=0;	//当前堆栈中的数字数量（不包括当前弹出的那个）
	int last_pop=MAX;	//上次弹出的数字
	int i;
	for(i=0;i<n;i++)
	{
		if(data[i]>stack_max)	//比之前最大数大，即要弹出比之前最大更大的数，需要压入数字
		{
			stack_num+=data[i]-stack_max-1;	//考虑容量是否够
			if(stack_num>=m)
				return false;
			stack_max=data[i];
			last_pop=data[i];
		}
		else if(data[i]<last_pop)	//比之前最大数大但比上一个弹出的小，即继续弹出
		{
			last_pop=data[i];
			stack_num--;
		}
		else	//比之前最大数大但比上一个弹出的大，不符合要求
			return false;
	}
	return true;
}