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
思路:
就是简单的模拟,比起网上的有些代码来说复杂了点。
具体操作见代码注释。
AC代码:
#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <stack>
using namespace std;
int main()
{
stack<int>st;
int n;
int a[10005];
scanf("%d",&n);
for(int i=0; i<n; i++)
scanf("%d",&a[i]);
int t;
int b[10005];
scanf("%d",&t);
while(t--)
{
//一定要清空栈。
while(!st.empty())
st.pop();
for(int i=0; i<n; i++)
scanf("%d",&b[i]);
//标记变量。
int flag = -1;
//关键代码。
int j = 0;
//从t组数据下手,这里的是用b[]储存的。
//a[]储存的是最开始的n个数据。
/*然后从b[]中的第一个数据开始比较,a[]中与b[]不相同的入栈(a[]),直到遇到相同的为止,此时不
入栈,而是索引直接往下走,继续找与b[]中第二个数相同的,截止到a[]全部比较完为止。
a[]比较完之后,再比较栈中剩余的数据和b[]剩余没有比较的数据是否相同?相同代表是合法的序列,
否则,就不是。
*/
for(int i=0;i<n;i++)
{
//a[]和b[]比较,不同的入栈。
while(a[j]!=b[i]&&j<n)
st.push(a[j++]);
if(j==n)
{
//a[]比较完了,标记一下。用于后面b[]剩余的数据进行比较。
flag = i;
break;
}
//相同,但是不入栈,索引直接往后走。
if(a[j]==b[i])
j++;
}
if(flag!=-1)
{
int k = flag;
//剩余的数据进行一一比较。
while(!st.empty())
{
if(st.top()!=b[k++])
{
printf("no\n");
break;
}
//注意出栈。
st.pop();
}
//全部出栈代表全部一致。
if(st.empty()) printf("yes\n");
}
/*这句话代表着比较完之后,栈内没有元素。比如:
a[] = {1,2,3,4,5}
b[] = {1,2,3,4,5}
顺序一致,栈内没有元素剩余。
*/
else printf("yes\n");
}
return 0;
}