题目翻译:
给定一个最多能存 M 个数字的栈,将 1∼N 按顺序压入栈中,过程中可随机弹出栈顶元素。
当 N 个数字都经历过入栈和出栈后,我们按照元素出栈的顺序,可以得到一个弹出序列。
现在给定一系列 1∼N 的随机排列序列,请你判断哪些序列可能是该栈的弹出序列。
例如,当 N=7,M=5 时,1, 2, 3, 4, 5, 6, 7可能是该栈的弹出序列,而 3, 2, 1, 7, 5, 6, 4 不可能是该栈的弹出序列。
题解思路:
这题是一道栈模拟的题目,每次模拟操作只有两个选择:
- 推入元素,即如果当前栈顶元素不等于当前需要输出的元素,则推入新的元素。
- 弹出元素,即如果当前栈顶元素等于当前需要输出的元素,则弹出栈顶元素,直至不相等为止。
而判断该弹出序列是否满足要求同样有两个依据:
- 栈中元素不能大于 m 。
- 做完所有操作后,栈中不能有元素即所有栈顶元。
代码:
#include<bits/stdc++.h>
using namespace std;
const int max_l = 1010;
int N, M, K;
int a[max_l], b[max_l];
int check()
{
stack<int> p;
int tag = 0;
for (int i = 0;i < M;i++)
{
if (p.empty()) p.push(a[i]);
else {
if (p.top() != b[tag] && p.size() < N)
p.push(a[i]);
}
while (p.size() && p.top() == b[tag]) p.pop(), tag++;
}
return tag == M;
}
int main()
{
cin >> N >> M >> K;
for (int i = 0;i < M;i++) a[i] = i + 1;
for (int i = 0;i < K;i++)
{
for (int i = 0;i < M;i++) cin >> b[i];
if (check()) cout << "YES" << endl;
else cout << "NO" << endl;
}
}
坑点:
无