1051 Pop Sequence(25 分)
作者: CHEN, Yue
单位: 浙江大学
时间限制: 400 ms
内存限制: 64 MB
代码长度限制: 16 KB
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以这样的顺序出栈。
那么其实我们可以通过模拟整个过程来进行判断,记 j = 1 为当前将要被压入栈的数字,用 i 来代表输入序列的下标:
接下来就会产生非常明显的两种情况,
1.栈为空时,我们将 j++ 压入栈
2.栈不为空时,执行一定判断
执行判断其实就是将栈顶元素和当前的序列元素进行比较:
1.如果序列元素大于栈顶元素,那么我们就可以接着将 j++ 压入栈进行下一次循环
2.如果序列元素小于栈顶元素,这个时候我们可以确定不能以这个顺序出栈了,因为如果栈顶元素大于序列元素,就证明需要输出的序列元素就在栈内,而想要输出这个元素必须先把栈顶的元素输出,此时就不是我们想要的序列,所以也就没有办法以这个顺序出栈了。
3.两者相等,这个是最令人高兴的情况,此时我们将栈顶元素出栈,并且 ++i 检查下一个序列元素
同时不要忘记题目中给定的栈容量和其他限制条件,当stack.size() > 最大容量的时候,也就代表着不能以这个顺序出栈了,造成同样结果的还有什么呢,就是 j > n + 1的时候(为什么是 n + 1 不是 n 呢,因为最后一次吧 j 压入之后还会对 j 进行 +1操作,所以 j 一定会大于 n 但是不大于 n + 1)。
那结束条件是什么呢,就是当检查完序列中最后一个元素的时候,就可以结束了,也就是 i >= n 的时候即返回 true.
AC代码:
#include <cstdio>
#include <stack>
using namespace std;
int se[1010];
bool correct(int, int);
int main() {
stack<int> is;
int m, n, k;
scanf("%d%d%d", &m, &n, &k);
for(int i = 0; i < k; ++i) {
if(correct(m, n)) {
printf("YES\n");
} else {
printf("NO\n");
}
}
return 0;
}
bool correct(int m, int n) {
stack<int> is;
for(int i = 0; i < n; ++i) {
scanf("%d", se + i);
}
int i = 0, j = 1;
while(1) {
if(is.empty()) {
is.push(j++);
} else {
if(is.top() < se[i]) {
is.push(j++);
} else if(is.top() > se[i]) {
return false;
} else {
is.pop();
++i;
}
}
if(is.size() > m) {
return false;
}
if(i >= n) {
return true;
}
if(j > n + 1) {
return false;
}
}
}
如有错误,欢迎指摘。