1051 Pop Sequence (25分)
给定一个最多能保存M个数的堆栈。按1,2,3,…,N的顺序推N个数字,然后随机弹出。您应该知道给定的数字序列是否可能是堆栈的弹出序列。例如,如果M是5,N是7,我们可以从堆栈中获得1,2,3,4,5,6,7,但不能从3,2,1,7,5,6,4。
输入规格:
每个输入文件包含一个测试用例。对于每一种情况,第一行包含3个数字(不超过1000个):M(堆栈的最大容量)、N(推送序列的长度)和K(要检查的POP序列的数目)。接下来是K行,每行包含N个数字的pop序列。一行中的所有数字都用空格隔开。
输出规格:
对于每个弹出序列,如果确实是堆栈的可能弹出序列,则在一行中打印“是”,如果不是,则打印“否”。
样本输入:
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一次入栈,在入栈的过程中 如果入栈的元素加好等于出栈序列当前等待出栈的元素,那么就让栈元素出栈,同时把出栈序列当前等待出栈的元素位置标记后移1位;
注:1,栈的大小有限 2.对栈操作前要判空 3.使用栈之前要清空栈
#include <stdio.h>
#include <stack>
using namespace std;
stack<int> s;
int main(){
int m,n,k;
scanf("%d%d%d",&m,&n,&k);
int num[2020];
while(k--){
for(int i=1;i<=n;i++){//下标从1开始
scanf("%d",&num[i]);
}
while(!s.empty()){//使用栈之前先清空栈
s.pop();
}
int current=1;//指向出栈序列中的带出栈的元素
int ans=1;
for(int i=1;i<=n;i++){
s.push(i);
if(s.size()>m){//超出容量,排序非法
ans=0;
break;
}//非空栈且栈顶元素与出栈序列当前位置的元素相同 ,循环出栈
while(!s.empty()&&s.top()==num[current]){
s.pop();
current++;
}
}
if(ans==1&&s.empty()){
printf("YES\n");
}else{
printf("NO\n");
}
}
return 0;
}