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
Hint
思路:根据出栈顺序模拟一遍出栈。
-
设置一个stack栈来模拟出栈,将入栈元素按顺序存入队列in中,将出栈元素按顺序存入队列out中。
-
从队列out中取元素e,判断stack是不是空。
a.stack为空,从队列in中取元素入stack栈。
b.stack非空,判断e和栈顶元素是否相等。如果相等,栈顶元素出栈;否则,继续从in中取元素入栈。 -
当in队列为空,循环结束,判断栈是否为空。如果栈为空,出栈顺序合法;否则,不合法。
#include <iostream>
using namespace std;
int main()
{
int n;
cin >> n;
int in[10000];
for (int i = 0; i < n; i++)
{
cin >> in[i];
}
int t;
cin >> t;
while (t--)
{
int out[10000];
for (int i = 0; i < n; i++)
{
cin >> out[i];
}
int stack[10000];
int *top = stack;
int j = 0;
bool flag = true;
for (int i = 0; i < n; i++)
{
int temp = out[i];
//对于出栈元素temp不等于栈顶元素or栈是空的情况,
//就从入栈顺序的数组in中读取下一个入栈元素,直到stack的栈顶元素和temp相等or数组in读取完毕。
if (top == stack || *(top-1) != temp)
{
for (; j < n;)
{
if (in[j] != temp)
{
*(top++) = in[j++];
}
else
{
*(top++) = in[j++];
break;
}
}
}
if (*(top-1) == temp)
{
top--;
}
else
{
flag = false;
break;
}
}
if (!flag)
{
cout << "no" << endl;
}
else
cout << "yes" << endl;
}
return 0;
}