Given a stack which can keep M M M numbers at most. Push N N N numbers in the order of 1 , 2 , 3 , . . . , N 1, 2, 3, ..., N 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 M M is 5 and N N 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.
Each input file contains one test case. For each case, the first line contains 3 numbers (all no more than 1000): M M M (the maximum capacity of the stack), N N N (the length of push sequence), and K K K (the number of pop sequences to be checked). Then K K K lines follow, each contains a pop sequence of N N N numbers. All the numbers in a line are separated by a space.
For each pop sequence, print in one line “YES” if it is indeed a possible pop sequence of the stack, or “NO” if not.
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
模拟栈操作,不过这个栈的大小有限制,输入序列限定为 1 → N 1\rightarrow N 1→N,通过模拟出栈入栈操作,看能否得到预期输出序列。
#include <iostream>
#include <cstdio>
#include <stack>
using namespace std;
int main(){
int m, n, k;
int a[1001];
stack<int> sta;
scanf("%d%d%d", &m, &n, &k);
while(k--){
for(int i=0; i<n; i++)
scanf("%d", a+i);
int t = 1, curi = 0;
while(t <= n){
if(sta.empty() || sta.top() != a[curi]){
if(sta.size() == m)
break;
sta.push(t);
//printf("入栈%d\n", t);
t++;
}
else{
// printf("出栈%d\n", sta.top());
sta.pop();
curi++;
}
}
while(!sta.empty() && sta.top() == a[curi]){
//printf("出栈%d\n", sta.top());
sta.pop();
curi++;
}
if(sta.empty())
printf("YES\n");
else
printf("NO\n");
while(!sta.empty())
sta.pop();
}
return 0;
}