且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

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

更新时间:2022-02-18 16:49:33

题目描述

输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如序列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));
    }
}