21、栈的压入、弹出序列

2017-09-02  本文已影响0人  quiterr

题目描述
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列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;
    }
}
上一篇 下一篇

猜你喜欢

热点阅读