剑指 Offer !! 31. 栈的压入、弹出序列

剑指 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 之前弹出。

思路:
创建一个栈,按照pushed的顺序压入元素到栈中;
再用一个指针放在popped序列的左端;
每次压入一个元素进栈后,比较一下此时栈顶和指针所指元素是否相等:
如相等,那么弹出栈并且指针前进一步;
如不等,那么把pushed的下一个元素压入栈,重复上述步骤。
代码如下:

class Solution {
    public boolean validateStackSequences(int[] pushed, int[] popped) {
        if(pushed==null ||pushed.length==0 ){
            return true;
        }
        int i=0;
        Stack<Integer> stack = new Stack<>();
        for(int cur:pushed){
            stack.push(cur);
            while(!stack.isEmpty()&&stack.peek()==popped[i]){
                stack.pop();
                i++;
            }
        }
        return stack.isEmpty();
    }
}

思考:
我觉得有一处设计得很巧妙:一个数组(这里是pushed数组)用for控制住,另一个数组用while循环。
如果写两个while循环的话,估计会很乱,……起码我写的很乱。

你可能感兴趣的:(算法,leetcode,数据结构)