剑指 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 的排列。

---

解法：栈顶匹配

思路

1. 用两个指针index1, index2分别代表pushed数组与popped数组当前元素的下标，从0开始
2. 将pushed[index1]入栈，查看当前栈顶元素是否和popped[index2]相等，如果相等则出栈,index2继续往后走一个，直到栈顶元素和popped[index2]不相等为止
3. 重复2，直到index1 = n（数组长度），停止循环
4. 如果当前栈为空，代表已经匹配完成，如果不为空，就代表不能匹配这个出栈顺序

特殊情况

当数组为空或者数组长度为0时，满足条件

代码

public boolean validateStackSequences(int[] pushed, int[] popped){
    if(pushed == null || pushed.length == 0) return true;
    Stack<Integer> stack = new Stack<>();
    int n = pushed.length;
    // index1和index2分别表示pushed和popped当前下标
    int index1 = 0, index2 = 0;
    // 遍历pushed数组
    while (index1 < n){
        // 将下一个数入栈
        stack.add(pushed[index1]);
        // 如果当前栈顶元素刚好a和popped数组当前元素b相等，则出栈，直到a,b不等为止
        while (!stack.isEmpty() && stack.peek() == popped[index2]){
            stack.pop();
            ++index2;
        }
        // 继续往后遍历
        ++index1;
    }
    // 已经全部出栈，就代表所有的已经匹配好了
    return stack.isEmpty();
}

提交结果

记得在大二学《数据结构》的时候，给出n个元素的数组，出栈顺序的种数有：