题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
思路
使用辅助栈
- 用辅助栈按压入顺序开始把元素弹进栈(不是一次性全压进去,边比较边压栈)
- 然后将弹出序列元素从头开始,依次与栈顶元素比较
- 一直入栈,直到栈顶能等于弹出序列第一元素,就弹出栈顶,弹出下标+1,和下一个弹出序列的元素继续进行比较,直到栈为空,说明比较结束了,弹出序列每个元素都符合弹出条件
若栈顶和弹出序列即将访问到的元素不等,则说明无法弹出该序列
实现
import java.util.Stack;
public class Solution {
public boolean IsPopOrder(int [] pushA,int [] popA) {
//用辅助栈按压入顺序开始把元素弹进栈(不是一次性全压进去,边比较边压栈)
//然后将弹出序列元素从头开始,依次与栈顶元素比较
//若当前待弹出元素==当前栈顶,就弹出栈顶,弹出下标+1。若不等,则说明无法弹出该序列
//特殊情况
if(pushA.length==0 || popA.length==0) return false;
Stack<Integer> stack=new Stack<>();
int popIndex=0; //当前弹出序列的下标
//按压入顺序开始把元素弹进栈,一直弹进,直到栈顶等于弹出序列第一个元素
for(int i=0;i<pushA.length;i++){
stack.push(pushA[i]);
//栈不为空前提下:比较当前弹出序列位置上的值和栈顶元素值是否相等
while(!stack.isEmpty() && popA[popIndex]==stack.peek()){
stack.pop(); //弹出序列popIndex位置上的元素已满足弹出条件
popIndex++; //继续判断下一个弹出元素
}
//如果不满足上述循环条件,就继续按压入顺序把元素弹进栈
}
return stack.isEmpty(); //最后若栈空,说明弹出序列每个元素都符合
}
}
出栈时使用while,不用if是因为一次性可能有多个元素可以出栈!