剑指offer系列之二十：栈的压入、弹出序列

题目描述

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

要判断一个序列是不是另一个序列的弹出序列，首先我们得知道栈的特点是FIFO，其次，由于压入序列是确定的（这里指的“确定”是说相对位置是确定的），就是说压入序列之间的元素可能会经历压入后马上被弹出的情况。具体思路是：根据弹出序列的第一个值，判断在该元素之前被压入栈的所有元素，并对压入栈的栈顶元素与弹出序列的第一个值进行比较，如果相等，则继续判断下一个元素，继续判断该元素之前需要压入栈中的所有元素，仍然进行比较，如果相等，继续判断下一个元素；如果所有的元素都被压入栈中，但是两者仍然不相等，说明弹出序列不可能是原序列的弹出序列。这样一直判断直到弹出序列的最后一个元素。

那么根据这种思路，可以写出如下代码（已被牛客AC）：

package com.rhwayfun.offer;

import java.util.ArrayList;

public class StackPushPopSequence {

    public boolean IsPopOrder(int[] pushA, int[] popA) {
        boolean possible = false;
        ArrayList<Integer> data = new ArrayList<Integer>();
        if (pushA.length > 0 && popA.length > 0 && pushA.length == popA.length) {
            int len = pushA.length;
            //当前元素
            int pop = 0;
            //下一个弹出的元素
            int nextPop = 0;

            int push = 0;
            //下一个压入的元素
            int nextPush = 0;

            int index = -1;
            while(nextPop - pop < len){
                //如果弹出的元素与第一个元素不相等就一直添加元素直到相等
                while(data.size() == 0 || data.get(index) != popA[nextPop]){
                    //当全部元素添加完毕之后，结束循环
                    if(nextPush - push == len) break;
                    data.add(pushA[nextPush]);
                    index++;
                    nextPush++;
                }

                if(data.get(index) != popA[nextPop]) break;
                //移除第一个比较的元素
                data.remove(index--);
                nextPop++;
            }
            //在集合全部元素比较完毕且弹出序列也比较完毕，返回true
            if(data.size() == 0 && nextPop - pop == len) possible = true;
        }
        return possible;
    }

    public static void main(String[] args) {
        int[] pushA = new int[]{1,2,3,4,5};
        int[] popA = new int[]{4,5,3,1,2};
        System.out.println(new StackPushPopSequence().IsPopOrder(pushA, popA));
    }
}