一、题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1、2、3、4、5是某栈的压栈序列,序列4、5、3、2、1是该压栈序列对应的一个弹出序列,但4、3、5、1、2就不可能是该压栈序列的弹出序列。
二、解题思路
如果下一个弹出的数字刚好是栈顶数字,那么直接弹出。如果下一个弹出的数字不在栈顶,我们把压栈序列中还没有入站的数字压入辅助栈,直到把下一个需要弹出的数字压入栈顶为止。如果所有的数字都压入栈了仍然没有找到下一个弹出的数字,那么该学列不可能是一个弹出序列。
三、Java代码实现
import java.util.Stack;
public class IsPopOrder {
public static void main(String[] args){
int[] push = {1, 2, 3, 4, 5};
int[] pop = {4, 5, 3, 2, 1};
int[] pop2 = {4, 3, 5, 1, 2};
System.out.println(isPop(push, pop));
System.out.println(isPop(push, pop2));
}
public static boolean isPop(int[] pushA, int[] popA){
Stack<Integer> sta = new Stack<Integer>();
int len = pushA.length;
int index = 0;
for(int i = 0; i < popA.length; i++){
if(sta.empty() || popA[i] != sta.peek()){
while(index < len && popA[i] != pushA[index]){
sta.push(pushA[index]);
index++;
}
if(index == len) return false;
else index++;//这一步容易忘
}
else sta.pop();
}
return true;
}