已知压栈序列，判断合法的弹出序列

问题描述：

输入两个整数序列，第一个表示栈的压入顺序，判断第二个序列是否为该栈的弹出序列（假设压栈的所有数字均不相等）。

例如：序列1,2,3,4,5是压栈序列，那么序列4,5,3,2,1就是一个合法的弹出序列，而4,3,5,1,2则不可能是合法的弹出序列。

问题出自《剑指offer》面试题22。

解题代码：

思路和书中讲诉类似：如果下一个弹出元素正好是栈顶元素，则直接弹出。否则把压栈序列中还没有入栈的元素依次入栈，直到栈顶元素等于弹出序列中要弹出的元素。如果所用元素都压栈了，并且栈顶元素任然不等于要弹出元素，则不合法。

代码实现上和书中有所区别：用两个下标i,j分别扫描压栈序列和弹出序列，循环的退出条件是：j < n && (i < n || !sta.empty())。退出循环时，如果弹出序列扫描完毕，则该弹出序列是合法的，否则不合法。

代码注意两点：

1，当i == n时，如果栈不为空，循环不能退出，还要继续执行循环体；

2，判断当前出栈元素是不是当前栈顶元素之前，要判断栈是否为空。如不判断，在栈为空时，执行取栈顶元素，程序异常。

代码如下：

#include <stdio.h>
#include <stdlib.h>
#include <stack>
using namespace std;

bool judge(int *a, int *b, int n) {
	stack<int> sta;

	int i = 1, j = 0;
	sta.push(a[0]);
	while (j < n && (i < n || !sta.empty())) {
		if (!sta.empty() && b[j] == sta.top()) {
			sta.pop();
			j++;
		} else {
			if (i < n) {
				sta.push(a[i]);
				i++;
			} else {
				return false;
			}
		}
	}

	if (j == n) return true;
	else return false;
}

int main() {
	int n, i;
	
	while (1) {
		scanf("%d", &n);
		int *a = new int[n];
		for (i = 0; i < n; i++) {
			scanf("%d", &a[i]);
		}

		int *b = new int[n];
		for (i = 0; i < n; i++) {
			scanf("%d", &b[i]);
		}

		bool rslt = judge(a, b, n);
		printf("%d\n", rslt);
	}
}