剑指Offer-31.栈的压入、弹出序列

题目描述

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

思路

用一个辅助栈

import java.util.ArrayList;
import java.util.Stack;

public class Solution {
    public boolean IsPopOrder(int [] pushA,int [] popA) {

        int len = pushA.length;
        if (len == 1)
            return pushA[0] == popA[0] ? true : false;

        Stack<Integer> stack = new Stack<>();

        boolean res = false;

        int i = 0;
        int j = 0;
        while (j < len) {

            // 如果弹出序列中当前要弹出的元素不在辅助栈栈顶
            while (i < len && (stack.isEmpty() || (stack.peek() != popA[j]))) {
                if (j == len - 1)
                    break;
                // 把压栈序列中未入栈的元素压入辅助栈，直到要弹出的元素在栈顶了为止
                stack.push(pushA[i]);
                i++;
            }

            if (stack.peek() != popA[j])
                break;

            stack.pop();
            j++;

        }

        if (stack.isEmpty() && j == len)
            res = true;

        return res;
    }
}