JZ21 栈的压入、弹出序列

原题链接

描述

---

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

示例1

---

输入：[1,2,3,4,5],[4,3,5,1,2]

返回值：false

思路

---

用一个栈模拟这个过程。

① 将pushA的第一个未入栈的元素入栈，然后判断栈顶元素是不是popA的第一个元素。

② 如果不是，则继续将pushA的下一个元素压入栈中，回到①；

③ 如果是，栈中元素出栈，回到①。

直到pushA所有元素都入栈。

最后判断栈中是否还有元素，没有的话说明出栈序列合法。

解答

---

import java.util.ArrayList;

public class Solution {
    public boolean IsPopOrder(int [] pushA,int [] popA) {
		if(pushA.length != popA.length||
          pushA.length == 0)
            return false;
        Stack<Integer> stack = new Stack<>();
        int index=0;
        for(int j : pushA){
            stack.push(pushA[j]);
            while(!stack.isEmpty() && stack.peek() == popA[i]){
                stack.pop();
                i++;
            }
        }
        return stack.isEmpty();
    }
}