https://www.acwing.com/problem/content/1537/
给定一个最多能存 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,N,K,分别表示栈的容量,数字个数,需要判断的序列个数。
接下来 K 行,每行包含一个 1∼N 的排列。
输出格式
对于每个序列,如果可能是该栈的弹出序列,则输出一行 YES,否则输出一行 NO。
数据范围
1≤M,N,K≤1000
输入样例:
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
#include<iostream>
#include<algorithm>
#include<stack>
using namespace std;
const int N=1010;
int in[N];//存储入栈序列
int out[N];//存储出栈序列
int main()
{
int n,m,k;
scanf("%d%d%d",&m,&n,&k);
for(int i=1;i<=n;i++)
{
in[i]=i;//入栈顺序为:1,2,3,4,5,,,,
}
while(k--)
{
stack<int> s;
for(int i=1;i<=n;i++)
{
scanf("%d",&out[i]);//输入出栈顺序
}
//思路:当栈s为空时,把入栈序列中的元素从前往后依次压入栈中
//若栈s不为空,则将栈顶元素和出栈序列中的当前的第一个元素比较,若相同,栈顶元素弹出,序列向后移,不相同就继续压入元素
int idex1=1,idex2=1;//idex1指向入栈元素的第一个,idex2指向出栈元素的第一个
int flag=1;
while(true)
{
if(s.empty())
{
if(idex1>n) break;//元素已经用完
s.push(in[idex1++]);
continue;
}
else if(s.top()==out[idex2])
{
while(s.top()==out[idex2])
{
s.pop();
idex2++;
if(s.empty()) break;
}
}
else if(s.top()!=out[idex2])//不相等就继续压入元素
{
s.push(in[idex1++]);
if(s.size()>m)//栈的元素个数超过指定大小
{
flag=0;
printf("NO\n");
break;
}
}
}
if(flag) printf("YES\n");
}
}