剑指Offer-题31(Java版):栈的压入与弹出序列

参考自:《剑指Offer——名企面试官精讲典型编程题》

题目:栈的压入与弹出序列
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。
例如:序列{1、2、3、4、5}是某栈的压栈序列,序列{4、5、3、2、1}是该压栈序列对应的一个弹出序列。但{4、3、5、1、2}就不可能是该压栈序列的弹出序列。

主要思路:借助辅助栈,并且使用一个指针指向弹出序列当前要弹出的元素。当每次把压栈序列的元素压入辅助栈中时,若辅助栈栈顶元素等于指针指向的元素,则弹出栈顶元素,并且将指针指向弹出序列的下一个元素,循环该过程,直到栈顶元素不等于指针指向的元素。最后,若压栈序列遍历完,同时辅助栈也为空(或者弹出序列也遍历完),则说明该弹出序列满足条件。

关键点:辅助栈,模拟压栈和弹出过程。

时间复杂度:O(n)

public class PopOrder
{
    public static void main(String[] args)
    {
        int[] pushA = {1, 2, 3, 4, 5};
        int[] popA = {4, 5, 3, 2, 1};
        int[] popB = {4, 3, 5, 1, 2};
        System.out.println(IsPopOrder(pushA, popA)); //true
        System.out.println(IsPopOrder(pushA, popB)); //false
    }

    private static boolean IsPopOrder(int[] pushA, int[] popA)
    {
        if (pushA.length == 0 || popA.length == 0)
            return false;
        Stack<Integer> helpStack = new Stack<>();
        int popIndex = 0;  //当前要弹出的元素位置
        for (int a : pushA)
        {
            helpStack.push(a);
            //辅助栈栈顶元素等于弹出序列当前要弹出的元素
            while (!helpStack.isEmpty() && popIndex < popA.length
                    && helpStack.peek() == popA[popIndex])
            {
                helpStack.pop();
                popIndex++;
            }
        }
        return helpStack.isEmpty();
    }
}

猜你喜欢

转载自blog.csdn.net/m0_37862405/article/details/80258275