面试题31：栈的压入、弹出序列

题目：

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

分析：

以弹出序列{4,5,3,2,1}为例分析压栈和弹出的过程。第一个希望被弹出的数字是4，因此4需要先压入辅助栈。压入栈的顺序由压栈序列确定，也就是在把4压入栈之前，数字1,2,3都需要先压入栈。此时栈里包含4个数字，分别是1,2,3,4，其中4位于栈顶。把4弹出栈后，剩下的3个数字是1,2,3。接下来希望弹出的数字是5，由于它不是栈顶数字，因此我们接着在第一个序列中把4以后的数字压入辅助栈，直到压入数字5。这时候5位于栈顶，就可以被弹出来了。接下来希望被弹出的3个数字一次是3,2,1。由于每次操作前它们都位于栈顶，因此直接弹出即可。

接下来再分析弹出序列{4,3,5,1,2}。第一个弹出的数字4的情况和前面的一样。把4弹出之后，3位于栈顶，可以直接弹出。接下来希望弹出的数字是5，由于5不是栈顶数字，到压栈序列里把没有压栈的数字压入辅助栈，直至遇到数字5。把数字5压入栈之后，5就位于栈顶了，可以弹出。此时站内有两个是最1和2，其中2位于栈顶。由于接下来需要弹出的数字是1，但1不在栈顶，我们需要从压栈序列中尚未压入栈的数字中搜索这个数字。但此时压栈序列中所有数字都已经压入栈了，所以该序列不是序列{1,2,3,4,5}对应的弹出序列。

总结：如果下一个弹出的数字刚好是栈顶数字，直接弹出；如果下一个弹出的数字不在栈顶，则把压栈序列中还没有入栈的数字压入辅助栈，直到把下一个需要弹出的数字压入栈为止；如果所有数字都压入辅助栈后仍然没有找到下一个弹出的数字，那么该序列不可能是一个弹出序列。

解法：

package com.wsy;

import java.util.Stack;

public class Main {
    public static void main(String[] args) {
        int[] push = new int[]{1, 2, 3, 4, 5};
        int[] pop = new int[]{4, 5, 3, 1, 2};
        System.out.println(check(push, pop));
    }

    public static boolean check(int[] push, int[] pop) {
        if (push == null || pop == null) {
            return false;
        }
        int length = push.length;
        Stack<Integer> stack = new Stack<Integer>();
        int popIndex = 0;
        for (int i = 0; i < length; i++) {
            stack.push(push[i]);
            while (!stack.empty() && stack.peek() == pop[popIndex]) {
                stack.pop();
                popIndex++;
            }
        }
        return popIndex == length;
    }
}