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