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
思路:
首先建立一个栈,将输入的数据保存在数组中,一个一个的进栈,然后每一个进栈的时候都与目标序列元素相比较,当二者相等的时候,出栈,然后继续获取栈顶元素余下一个序列元素相比较,相等就接着比较下一个,如果不相等就继续入栈操作,当入栈操作结束后。判断时候为空栈,如果为空栈,输出yes,否则输出no。
#include <stdio.h>
#include <string.h>
#include <math.h>
#include <stdlib.h>
#define maxsize 10000+10
typedef int element;
typedef struct
{
element *base;
element *top;
int stacksize;
} sqstack;
void initstack(sqstack &s)
{
s.base = (element *)malloc(sizeof(element) * maxsize);
s.top = s.base;
s.stacksize = maxsize;
}
void push(sqstack &s, int x)
{
*(s.top) = x;
s.top++;
}
element getop(sqstack &s)
{
return *(s.top-1);
}
int emptystack(sqstack &s)
{
if(s.base == s.top)
return 1;
else
return 0;
}
element pop(sqstack &s)
{
return *(--s.top);
}
int main()
{
int n, i, t, k;
int b[10000+10], a[10000+10];
memset(a, 0, sizeof(a));
sqstack s;
scanf("%d", &n);
for(i=0; i<n; i++)
{
scanf("%d", &a[i]);
}
scanf("%d", &t);
while(t--)
{
k = 0;
initstack(s);
memset(b, 0, sizeof(b));
for(i=0; i<n; i++)
scanf("%d", &b[i]);
for(i=0; i<n; i++)
{
push(s, a[i]);
while(getop(s) == b[k] && emptystack(s) != 1)
{
pop(s);
k++;
}
}
if(emptystack(s) == 1)
printf("yes\n");
else
printf("no\n");
}
return 0;
}