剑指 offer 31：栈的压入、弹出序列

剑指 offer 31：栈的压入、弹出序列

代码一

可以发现判断一个序列不是栈的弹出序列的规律：如果下一个弹出的数字刚好是栈顶数字，那么直接弹出；如果下一个弹出的数字不在栈顶，则把压栈序列中还没入栈的数字压入辅助栈，直到把下一个需要弹出的数字压入栈顶为止；如果所有数字都压入栈后仍没有找到下一个弹出的数字，那么该序列不可能是一个弹出序列。出栈为主考虑。

class Solution {
    public boolean validateStackSequences(int[] pushed, int[] popped) {
        
        if(pushed.length == 0 && popped.length==0){
            return true;
        }
        if(pushed.length==0 || popped.length==0){
            return false;
        }
        int i=0;
        int j=0;
        int len1 = pushed.length;
        int len2 = popped.length;
        Stack<Integer> s = new Stack<>();

        while(i<len1||j<len2){
            if(s.empty()){
                s.push(pushed[i]);
                i++;
            }
            //下一个弹出的数字刚好是栈顶数字，直接弹出
            if(popped[j]==s.peek()){
                j++;
                s.pop();
            }else{
                //下一个弹出的数字不在栈顶
                //把压栈序列中还没入栈的数字压入栈，直到把下一个需要弹出的数字压入栈顶为止
                while(i<len1&&popped[j]!=s.peek()){
                    s.push(pushed[i]);
                    i++;
                }
                if(popped[j]==s.peek()){
                    j++;
                    s.pop();
                }
                //如果所有数字都入栈仍然没找到下一个弹出的数字，那么该序列不可能是一个弹出序列
                else{
                    return false;
                }
            }

        }
        return true;

    }

时间复杂度$O(n)$，$n$是序列的长度，序列中的所有元素都需要入栈和出栈一次。
空间复杂度$O(n)$，需要一个辅助栈来判断压入、弹出序列是否正确。

代码二

更简洁
元素按照入栈顺序入栈，入栈之后判断栈顶元素是否和出栈序列的元素对应，如果对应则出栈，一直执行，直到不对应或者栈空，重新执行入栈操作，重复此步骤。入栈为主考虑。

class Solution {
    public boolean validateStackSequences(int[] pushed, int[] popped) {
        Stack<Integer> s = new Stack<>();
        int i=0;
        for(int num:pushed){
            s.push(num);
            while(!s.empty() && s.peek()==popped[i]){
                s.pop();
                i++;
            }
        }
        return s.empty();

    }
}