21、栈的压入弹出序列
题目描述:
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否可能为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
思路:
使用栈辅助模拟压栈和出栈过程。
import java.util.Stack;
public class Solution {
public boolean IsPopOrder(int [] pushA,int [] popA) {
//判空和错误
if(pushA==null||popA==null||pushA.length==0||popA.length==0||pushA.length!=popA.length){return false;}
Stack<Integer> st = new Stack<>();//辅助栈
st.push(pushA[0]);
int i=0,j=0;//i是压栈数组下标,j是出栈数组下标
while(j<popA.length){
if(popA[j]!=st.peek()){//栈顶元素和出栈数组当前剩余第一个元素不等时继续压栈
if(i>=pushA.length){//所有压栈数组元素已压栈但仍要压栈则说明出栈序列错误
return false;
}
st.push(pushA[i++]);//元素已压栈一个,故压栈数组下标+1
}
else{
st.pop();//栈顶元素和出栈数组当前剩余第一个元素相等时出栈
j++;//元素已出栈一个,故出栈数组下标+1
}
}
return true;
}
}
同思路另一种解法:
import java.util.Stack;
public class Solution {
public boolean IsPopOrder(int [] pushA,int [] popA) {
if(pushA==null||popA==null||pushA.length==0||popA.length==0||pushA.length!=popA.length){return false;}
Stack<Integer> st = new Stack<>();
int j=0;//j是出栈数组下标
for(int i=0;i<pushA.length;i++){
st.push(pushA[i]);
//当栈不为空且顶元素和出栈数组当前剩余第一个元素相等时出栈
while(!st.empty()&&st.peek()==popA[j]){
st.pop();
j++;
}
}
return st.empty();
}
}
注:这里要判断栈st
是否为空,因为可能连续出栈。