出栈序列的合法性
题目难度: ☀️
题目类型 : 栈
题目描述:
给定一个最大容量为 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
解题思路:
默认读者已经了解了栈,题目按数字顺序入栈,所以任何一个出栈得到的数字,那么比它小的数早已经入栈了。所以准备一个栈,遍历已经出栈的数组,小于等于当前数字的入栈,然后弹出这个数字。注意栈的最大容量。
以下是完整AC代码:
#include<iostream>
#include<stack>
#include<vector>
#include<algorithm>
using namespace std;
int main() {
int m, n, k; cin >> m >> n >> k;
while (k--) {
bool flag = true;
// flag表示出栈是否合法
vector<int> N(n);
// 整型数组存储出栈数字
for (int i = 0; i < n; i++) cin >> N[i];
stack<int> help; int t = 1;
// 创建help栈模拟入栈操作,t为从1到n入栈的数字
for (int i = 0; i < n && flag; i++) {
// 一旦出栈顺序不合法,直接跳出for循环
while (t <= N[i]) {
// 小于等于当前数字的入栈
help.push(t++);
}
if (help.top() == N[i] && help.size() <= m) help.pop();
// 入栈元素的数量不超过栈的最大容量m并且help栈顶元素等于当前数字弹出
else flag = false;
}
if (flag) cout << "YES";
else cout << "NO";
if (k != 0) cout << "\n";
// 注意输出格式,PTA要求严格
}
return 0;
}