劍指Offer-棧的壓入彈出序列Java

21、棧的壓入彈出序列

題目描述：
輸入兩個整數序列，第一個序列表示棧的壓入順序，請判斷第二個序列是否可能爲該棧的彈出順序。假設壓入棧的所有數字均不相等。例如序列1,2,3,4,5是某棧的壓入順序，序列4,5,3,2,1是該壓棧序列對應的一個彈出序列，但4,3,5,1,2就不可能是該壓棧序列的彈出序列。（注意：這兩個序列的長度是相等的）

思路：
使用棧輔助模擬壓棧和出棧過程。

import java.util.Stack;
public class Solution {
    public boolean IsPopOrder(int [] pushA,int [] popA) {
        //判空和錯誤
        if(pushA==null||popA==null||pushA.length==0||popA.length==0||pushA.length!=popA.length){return false;}
        Stack<Integer> st = new Stack<>();//輔助棧
        st.push(pushA[0]);
        int i=0,j=0;//i是壓棧數組下標，j是出棧數組下標
        while(j<popA.length){
            if(popA[j]!=st.peek()){//棧頂元素和出棧數組當前剩餘第一個元素不等時繼續壓棧
                if(i>=pushA.length){//所有壓棧數組元素已壓棧但仍要壓棧則說明出棧序列錯誤
                    return false;
                }
                st.push(pushA[i++]);//元素已壓棧一個，故壓棧數組下標+1
            }
            else{
                st.pop();//棧頂元素和出棧數組當前剩餘第一個元素相等時出棧
                j++;//元素已出棧一個，故出棧數組下標+1
            }
        }
        return true;
    }
}

同思路另一種解法：

import java.util.Stack;
public class Solution {
    public boolean IsPopOrder(int [] pushA,int [] popA) {
        if(pushA==null||popA==null||pushA.length==0||popA.length==0||pushA.length!=popA.length){return false;}
        Stack<Integer> st = new Stack<>();
        int j=0;//j是出棧數組下標
        for(int i=0;i<pushA.length;i++){
            st.push(pushA[i]);
            //當棧不爲空且頂元素和出棧數組當前剩餘第一個元素相等時出棧
            while(!st.empty()&&st.peek()==popA[j]){
                st.pop();
                j++;
            }
        }
        return st.empty();
    }
}

注：這裏要判斷棧st是否爲空，因爲可能連續出棧。