题目1366:栈的压入、弹出序列
时间限制:1 秒内存限制:32 兆特殊判题:否提交:2143解决:816
题目描述:
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。
输入:
每个测试案例包括3行:
第一行为1个整数n(1<=n<=100000),表示序列的长度。
第二行包含n个整数,表示栈的压入顺序。
第三行包含n个整数,表示栈的弹出顺序。
输出:
对应每个测试案例,如果第二个序列是第一个序列的弹出序列输出Yes,否则输出No。
样例输入:
5
1 2 3 4 5
4 5 3 2 1
5
1 2 3 4 5
4 3 5 1 2
样例输出:
Yes
No
注意不要使用stack,使用的内存超过题目限制,因为其是动态增长,内存复制,需要很大的内存空间。改为数组模拟栈。
#include <iostream>
#include<stdio.h>
#include<stack>
using namespace std;
bool isPopSequence(int* pPush,int* pPop,int nLength){
if(pPush==NULL||pPush==NULL||nLength<=0){
return false;
}
int* pPushNext = pPush;
int* pPopNext = pPop;
int pNext = 0;
int pTop = -1;
int* stackData = new int[nLength+1];
while(pPopNext-pPop<nLength){
while(pTop==-1||stackData[pTop]!=*pPopNext){
stackData[pNext++] = *pPushNext;
pPushNext++;
pTop++;
}
if(pTop>=0&&stackData[pTop]!=*pPopNext){
return false;
}
pNext--;
pTop--;
pPopNext++;
}
if(pTop==-1&&pPopNext-pPop==nLength){
return true;
}
return false;
}
int main()
{
int n;
while(scanf("%d",&n)!=EOF){
int* a = new int[n];
int* b = new int[n];
for(int i=0;i<n;i++){
scanf("%d",&a[i]);
}
for(int i=0;i<n;i++){
scanf("%d",&b[i]);
}
if(isPopSequence(a,b,n)){
printf("Yes\n");
}else{
printf("No\n");
}
}
return 0;
}