题目
-
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等
-
例如,序列 {1,2,3,4,5} 是某栈的压栈序列,序列 {4,5,3,2,1} 是该压栈序列对应的一个弹出序列,但 {4,3,5,1,2} 就不可能是该压栈序列的弹出序列
解法
/**
* 判断入栈、出栈顺序
*
* @param push
* @param pop
* @return
*/
public static boolean isValidStackOrder(int[] push, int[] pop) {
if (push.length == 0 || pop.length == 0) return false;
Stack<Integer> mStack = new Stack<>();
int index = 0;
for (int i = 0; i < push.length; i++) {
mStack.push(push[i]);
// 栈顶元素与弹出元素相同,弹出
while (!mStack.isEmpty() && mStack.peek() == pop[index]) {
mStack.pop();
index++;
}
}
return mStack.isEmpty();
}
main方法测试
public static void main(String[] args) {
int[] arrOne = {1, 2, 3, 4, 5};
int[] arrTwo = {4, 5, 3, 2, 1};
int[] arrThree = {4, 3, 5, 1, 2};
System.out.println(isValidStackOrder(arrOne, arrTwo));
System.out.println(isValidStackOrder(arrOne, arrThree));
}
日志输出
true
false