题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列1,2,3,4,5是某栈的压入顺序,序列4,5,3,2,1是该压栈序列对应的一个弹出序列,但4,3,5,1,2就不可能是该压栈序列的弹出序列。(注意:这两个序列的长度是相等的)
这题如果知道思路的话挺简单的,但是第一次做忽略了栈为空的情况,不处理会抛出异常,这一点跟Queue不一样,Queue如果为空,peek函数会返回null。
import java.util.*;
public class Solution {
public boolean IsPopOrder(int [] pushA,int [] popA) {
Stack<Integer> stack = new Stack<>();
int j=0;//pushA的索引
//逐个判断popA的元素是不是弹出元素
for(int i=0; i<popA.length; i++){
//如果栈为空
if(stack.isEmpty()){
if(j<pushA.length){
stack.push(pushA[j]);
j++;
}else{
return false;
}
}
//栈顶元素不是弹出元素,就继续入栈
while(stack.peek()!=popA[i]){
if(j<pushA.length){
stack.push(pushA[j]);
j++;
}else{//pushA的元素都已入栈,仍没有找到popA的相应元素
return false;
}
}
//找到元素之后弹出,以便继续判断下一个元素
stack.pop();
}
return true;
}
}