栈的压入、弹出序列

题目描述：

输入两个整数序列，第一个序列表示栈的压入顺序，请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序，序列4，5,3,2,1是该压栈序列对应的一个弹出序列，但4,3,5,1,2就不可能是该压栈序列的弹出序列。

import java.util.Stack;

public class Stk {

	public static void main(String[] args) {
		int[] a = new int[] { 1, 2, 3, 4, 5 };
		int[] b = new int[] { 4, 5, 3, 2, 1 };
		System.out.println(match(a, b));
		int[] c = { 4, 3, 5, 1, 2 };
		int[] d = {1,2,3,4,5};
		int[] e={5,4,3,2,1};
		System.out.println(match(a, c));
		System.out.println(match(a, d));
		System.out.println(match(a, e));
	}

	/**
	 * 检查a是否是b的弹出序列
	 * 
	 * @param a
	 * @param b
	 * @return
	 */
	public static boolean match(int[] b, int[] a) {
		if (a.length != b.length) {
			return false;
		}
		Stack<Integer> stk = new Stack<>();
		int index = -1;
		for (int i = 0; i < a.length; i++) {
			if (stk.isEmpty() || a[i] != stk.peek()) {
				for (int j = index + 1; j < b.length; j++) {
					stk.push(b[j]);
					if (a[i] == b[j]) {
						index = j;
						stk.pop();
						break;
					}
				}
			} else {
				stk.pop();
			}
		}
		return stk.isEmpty();
	}

}