剑指 offer面试题22 栈的压入和弹出序列

题目描述

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

思路

栈顶元素与出栈序列的第一个元素进行比较，如果相等，则出栈。如果不相等或者栈为空则继续入栈，如果最后栈为空了，则说明给定的弹出序列是压栈序列的一个弹出序列。

代码

import java.util.Stack;

public class Solution {
    public boolean IsPopOrder(int [] pushA,int [] popA) {
        if(pushA == null || popA == null) return false;
        if(pushA.length == 0 || popA.length == 0) return false;
        
        Stack<Integer> stack = new Stack<>();
        
        int index = 0;
        
        for(int i = 0; i < pushA.length; i++) {
            stack.push(pushA[i]);
            while(!stack.isEmpty() && popA[index] == stack.peek()) {
                index++;
                stack.pop();
            }
        }
        
        return stack.isEmpty() && (index == popA.length);
    }
}