力扣 剑指 Offer 31. 栈的压入、弹出序列 中等

题目

输入两个整数序列，第一个序列表示栈的压入顺序，请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如，序列 {1,2,3,4,5} 是某栈的压栈序列，序列 {4,5,3,2,1} 是该压栈序列对应的一个弹出序列，但 {4,3,5,1,2} 就不可能是该压栈序列的弹出序列。

示例 1：

输入：pushed = [1,2,3,4,5], popped = [4,5,3,2,1]
输出：true
解释：我们可以按以下顺序执行：
push(1), push(2), push(3), push(4), pop() -> 4,
push(5), pop() -> 5, pop() -> 3, pop() -> 2, pop() -> 1
示例 2：

输入：pushed = [1,2,3,4,5], popped = [4,3,5,1,2]
输出：false
解释：1 不能在 2 之前弹出。

提示：

0 <= pushed.length == popped.length <= 1000
0 <= pushed[i], popped[i] < 1000
pushed 是 popped 的排列。

题解

按照我们平常判断是否是出栈序列的想法来写。

先通过入栈序列不断入栈，同时与出栈序列进行比较，相同的就不需要入栈，也与栈顶元素进行比较，相同的就出栈，最后出栈序列再与站内元素进行比较，如果最后栈内为空则为true，否则为false。

class Solution {
    public boolean validateStackSequences(int[] pushed, int[] popped) {
        Deque<Integer> st = new ArrayDeque();
        int len1 = pushed.length, len2 = popped.length;
        if(len1 != len2) return false;
        if(len1 == 0) return true;
        int i = 0, j = 0;
        while(i < len1){   // 不断入栈
            if(pushed[i] == popped[j]){  //入栈元素和出栈元素一样
                i++;
                j++;
            }
            else if(!st.isEmpty() && st.peek() == popped[j]){  // 栈顶元素等于出栈元素
                st.pop();
                j++;
            }
            else{  //入栈元素、栈顶元素和出栈元素不一样
                st.push(pushed[i]);
                i++;
            }
        }
        while(j < len2){   // 比较栈内元素和出战队列内剩余元素
            if(st.peek() == popped[j]){
                st.pop();
                j++;
            }
            else
                return false;
        }
        if(st.isEmpty()) return true;
        return false;
    }
}

简洁代码

class Solution {
    public boolean validateStackSequences(int[] pushed, int[] popped) {
        Deque<Integer> st = new ArrayDeque();
        int j = 0;
        for(int elem : pushed){  // 遍历进栈序列
            st.push(elem);   // 元素进栈
            while( j < popped.length && !st.isEmpty() && st.peek() == popped[j]){  //栈顶元素等于出栈序列首部
                st.pop();  //元素出栈
                j++;
            }
        }
        return j == popped.length;   // 看出站列表指针是否指向出战列表末尾
    }
}