-
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列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
解题分析:
如果下一个弹出的数字刚好是栈顶数字,那么直接弹出。如果下一个弹出的数字不在栈顶,我们把压栈序列中还没有入栈的数字压入辅助栈,直到把下一个需要弹出的数字压入栈顶为止。如果所有的数字都压入了栈了仍然没有找到下一个弹出的数字,那么该序列不可能是一个弹出序列。
代码如下:
#include <iostream>
#include <stack>
using namespace std;
bool isPopSequence(int* pushSeq,int *popSeq,int length){
if(pushSeq==NULL || popSeq==NULL || length<1)
return false;
int ipop=0,jpush=0;
stack<int> istack;
while(ipop<length){
if(istack.empty() || istack.top()!=popSeq[ipop]){ //栈顶元素不等于当前栈顶元素,则遍历压入序列
while(jpush<length && pushSeq[jpush]!=popSeq[ipop]){
istack.push(pushSeq[jpush]);
++jpush;
}
if(jpush==length)
return false;
else {
/*istack.push(pushSeq[jpush]);
++jpush;
istack.pop();*/ //是等价的
++jpush;
}
}else
istack.pop();//栈顶元素等于当前弹出序列元素,则直接弹出
++ipop;//下一个弹出元素
}//while循环正常结束,则代表是弹出序列
return true;
}
void TestProgram22(){
int length=0;
while(cin>>length){
int *popSeq=new int[length];
int *pushSeq=new int[length];
for(int i=0;i<length;i++)
cin>>pushSeq[i];
for(int i=0;i<length;i++)
cin>>popSeq[i];
if(isPopSequence(pushSeq,popSeq,length))
cout<<"Yes"<<endl;
else
cout<<"No"<<endl;
}
}
int main(){
TestProgram22();
return 0;
}