数据结构系列内容的学习目录 → \rightarrow →浙大版数据结构学习系列内容汇总。
题目描述: 给定一个最多能保存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;
}
测试: 输入样例的测试效果如下图所示。