数据结构实验之栈与队列七:出栈序列判定
Time Limit: 30 ms Memory Limit: 1000 KiB
Problem Description
给一个初始的入栈序列,其次序即为元素的入栈次序,栈顶元素可以随时出栈,每个元素只能入栈依次。输入一个入栈序列,后面依次输入多个序列,请判断这些序列是否为所给入栈序列合法的出栈序列。
例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个出栈序列,但4,3,5,1,2就不可能是该序列的出栈序列。假设压入栈的所有数字均不相等。
Input
第一行输入整数n(1<=n<=10000),表示序列的长度。
第二行输入n个整数,表示栈的压入顺序。
第三行输入整数t(1<=t<=10)。
后面依次输入t行,每行n个整数,表示要判断的每一个出栈序列。
Output
对应每个测试案例输出一行,如果由初始入栈序列可以得到该出栈序列,则输出yes,否则输出no。
Sample Input
5 1 2 3 4 5 2 4 5 3 2 1 4 3 5 1 2
Sample Output
yes no
#include<stdio.h>
#include<string.h>
#include <stack>
using namespace std;
const int maxn = 1e5+10;
stack<int>q;
int main()
{
int T, n;
int a[maxn];
int b[maxn];
scanf("%d", &n);
for(int i=0; i<n; i++)
{
scanf("%d", &a[i]);
}
scanf("%d", &T);
while(T--)
{
int i, j;
while(!q.empty()) q.pop();
for(int i=0; i<n; i++)
scanf("%d", &b[i]);
i=0;
j=0;
while(j<n)//只会有三种情况,1、之间出栈(压根就没进)。2、进来了再出,3、进栈
{
if(a[i]==b[j])
{
i++;
j++;
}
else if(!q.empty()&&q.top()==b[j])
{
q.pop();
j++;
}
else if(i<n)
{
q.push(a[i]);
i++;
}
else break;
}
if(j>=n)
printf("yes\n");
else
printf("no\n");
}
return 0;
}