LeetCode 946. 验证栈序列

Posted 2021-05-20 数据结构和算法

想看更多算法题，可以扫描上方二维码关注我微信公众号“数据结构和算法”，截止到目前我已经在公众号中更新了500多道算法题，其中部分已经整理成了pdf文档，截止到目前总共有1000多页（并且还会不断的增加），可以在公众号中回复关键字“pdf”即可下载。

使用栈来解决

原理比较简单，我们来看下代码

public boolean validateStackSequences(int[] pushed, int[] popped) {
    //创建一个栈
    Stack<Integer> stack = new Stack<>();
    //记录popped数组访问到哪一个元素了
    int index = 0;
    //遍历pushed数组中的所有元素
    for (int num : pushed) {
        stack.push(num);//把当前元素push到栈中
        while (!stack.empty() && stack.peek() == popped[index]) {
            //如果栈顶元素等于popped[index]，就让栈顶元素出栈
            stack.pop();
            index++;
        }
    }
    return stack.empty();
}

使用单个指针

我们还可以只用单个变量i来结合pushed数组来模拟栈的存储，原理和上面类似，来看下代码

public boolean validateStackSequences(int[] pushed, int[] popped) {
    int i = 0;//相当于栈中元素的个数
    int j = 0;//记录popped数组访问到哪个元素了
    //遍历pushed数组中的所有元素
    for (int num : pushed) {
        pushed[i++] = num;//把当前元素在从新放到pushed数组中
        while (i > 0 && pushed[i - 1] == popped[j]) {
            // pushed[i - 1]类似于栈顶的元素
            --i;
            ++j;
        }
    }
    return i == 0;
}