【剑指offer】【九度oj】栈的压入压出

本文介绍了一种算法,用于判断给定的整数序列是否为特定压栈序列的有效弹出序列。通过使用辅助栈,文章提供了一个O(n)时间复杂度的解决方案,并附带了详细的实现代码。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目描述:

输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列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;
}





评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值