剑指offer31.栈的压入、弹出序列

栈的压入、弹出序列

输入两个整数序列，第一个序列表示栈的压入顺序，请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如，序列 {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 的排列。

来源：力扣（LeetCode） 链接：https://leetcode-cn.com/problems/zhan-de-ya-ru-dan-chu-xu-lie-lcof 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

法一：

使用一个栈来模拟即可。

对于相同的数字则弹出， 不同则加入到栈，观察最后，栈是否为空即可判断是否符合栈的规则。

public boolean validateStackSequences(int[] pushed, int[] popped) { if (pushed.length == 0) { return popped.length == 0; } if (pushed.length != popped.length) { return false; } Deque<Integer> stack = new ArrayDeque<>(); int popIdx = 0; for (int i = 0; i < popped.length; i++) { // 把当前的数据放入栈 stack.push(pushed[i]); // 如果相同则一直弹出 while (!stack.isEmpty() && popped[popIdx] == stack.peek()) { stack.pop(); popIdx++; } } return stack.isEmpty(); }

法二(法一优化版)：

因为法一是建立在不可以修改的基础上，故需要O(n)的空间，如果要把空间复杂度降低到O(1)，可以直接在原地修改。

就有了法二，其实方法和法一类似。

public boolean validateStackSequences(int[] pushed, int[] popped) { if (pushed.length == 0) { return popped.length == 0; } if (pushed.length != popped.length) { return false; } int pushIdx = 0; int popIdx = 0; for (int i = 0; i < popped.length; i++) { pushed[pushIdx] = pushed[i]; while (pushIdx != -1 && popped[popIdx] == pushed[pushIdx]) { pushIdx--; popIdx++; } pushIdx++; } return popIdx == popped.length; }