- 题目描述:
-
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列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
- 输出:
- 样例输入:
- 样例输出:
大体思路:我们可以建立一个辅助栈来存储入栈的部分序列,可以o(n)的时间复杂度内解决问题。出栈序列的元素首先与辅助栈(若不空)的栈顶元素比较,若不等再与入栈序列的元素比较。若与入栈序列的元素仍不相等,则当前入栈序列的元素进入辅助栈,出栈序列的当前元素与入栈序列的后面元素继续比较。若辅助栈的栈顶元素和入栈序列的全部元素都不与出栈序列的当前元素相等,则该序列不是一个正确的出栈序列;若所有出栈序列元素都比较完,则是一个正确的出栈序列。
代码如下:
#include <iostream>
#include <algorithm>
#include <string>
#include <stack>
using namespace std;
int a[100010],b[100010];
stack<int> s;
int main()
{
int n,i,j;
while(cin>>n)
{
for( i=0; i<n; i++ )
cin>>a[i];
for( i=0; i<n; i++ )
cin>>b[i];
i=j=0;
while( !s.empty() ) //这个很关键,因为有多个测试用例,所以需要清空上个测试用例的栈。
s.pop();
while(i<=n)
{
if( !s.empty() && b[j]==s.top() )
{
s.pop();
j++;
}
else
{
if( i==n )
break;
if( b[j]==a[i] )
{
i++;
j++;
}
else
{
s.push(a[i]);
i++;
}
}
}
if( j == n )
cout<<"Yes"<<endl;
else
cout<<"No"<<endl;
}
return 0;
}