数据结构（二）线性结构 —— 编程作业 04 ：Pop Sequence

数据结构系列内容的学习目录$\to$浙大版数据结构学习系列内容汇总。

  题目描述： 给定一个最多能保存M个数的堆栈。按1,2,3,…,N的顺序使得N个数字入栈，然后随机弹出，判断给定的数字序列是否可能是堆栈的pop序列。例如，如果M为5，N为7，我们可以从堆栈中得到1、2、3、4、5、6、7，但不能得到3、2、1、7、5、6、4。

  输入格式： 每个输入文件包含一个测试用例。
        对于每种情况，第一行包含3个数字(都不超过1000)：M(堆栈的最大容量)、N (push序列的长度)和K(需要检查的pop序列的数量)。
        然后是K行，每一行包含N个数字的pop序列。一行中的所有数字都用空格隔开。

  输出格式： 对于每个pop序列，如果它确实是堆栈中的pop序列，则打印一行"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

  代码实现：

#include<iostream>
using namespace std;
#include<vector>
#include<stack>

int main() 
{
	int m, n, k;
	cin >> m >> n >> k;
	for (int i = 0; i < k; i++) 
	{
		stack<int> s;
		vector<int> v(n + 1);
		for (int j = 1; j <= n; j++)  //输入需要判断的是否可能是堆栈出栈的序列
			cin >> v[j];
		int cur = 0;  //出栈的次数
		for (int j = 1; j <= n; j++)
		{
			s.push(j);  //将数字1,...,n按顺序依次压入堆栈
			if (s.size() > m)  //如果s的大小已经大于堆栈的容量，说明还没有元素跟需要判断的pop序列的首元素对应上而被弹出，这个pop序列不可能是堆栈的pop序列
				break;
			while (!s.empty() && s.top() == v[cur + 1])  //当s不为空且s的栈顶元素与需要判断的pop序列的元素对应上时，弹出此元素
			{
				s.pop();  //弹出
				cur++;  //出栈的次数+1
			}
		}
		if (cur == n)  //出栈次数等于push序列的长度，说明需要判断的pop序列的所有元素均已出栈，这个pop序列可能是堆栈的pop序列
			cout << "YES" << endl;
		else
			cout << "NO" << endl;
	}
	system("pause");
	return 0;
}

  测试： 输入样例的测试效果如下图所示。