数据结构实验之栈七：出栈序列判定

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。

Example Input

51 2 3 4 524 5 3 2 14 3 5 1 2

Example Output

yesno

Author

 

#include <bits/stdc++.h>
using namespace std;

int a[10001],b[10001];

int main()
{
int i,j,k,n,m,flag;

cin>>n;
for(i=0;i<n;i++)
cin>>a[i];
cin>>m;
while(m--)
{
stack <int> s;
j=0;
k=0;

for(i=0;i<n;i++)
cin>>b[i];
while(j<n)
{
if(a[k]==b[j])
{
k++;
j++;
}
else if(!s.empty()&&s.top()==b[j])
{
j++;
s.pop();
}
else if(k<n)
{
s.push(a[k]);
k++;
}
else if(i>=n)
{
j++;
}

}
if(s.empty())
cout<<"YES"<<endl;
else
cout<<"NO"<<endl;
}
return 0;
}