栈的压入、弹出序列

依然是舍友春招的笔试题之一。

题目重述

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

思路

构造一个辅助栈模拟第一个序列进栈，并且每次从辅助栈栈顶元素往下与第二个序列比较。如果相同则出栈。

#include <iostream>
using namespace std;
long long n;
long long a[100001]={0};
long long b[100001]={0};
void judgeOrder(long long);

int main()
{
	while(cin>>n)
	{
		a[0] = b[0] = n;
		for(int i = 1;i <= n; i++)
		{
			cin >> a[i];
		}
		for(int i = 1;i <= n; i++)
		{
			cin >> b[i];
		}
		judgeOrder(n);
	}
	return 0;
}

void judgeOrder(long long num)
{
	int top=0;
	int aIndex = 1;
	int bIndex = 1;
	int stack[100001]={0};
	stack[0] = num;
	for(int i = 0;i < num; i++)
	{
		stack[++top] = a[aIndex++];
		while(top>0 && stack[top] == b[bIndex])
		{
			top --;
			bIndex ++;
		}
	}
	if(top != 0)
		cout<<"No"<<endl;
	else
		cout<<"Yes"<<endl;
}