1. 程式人生 > >《劍指offer》系列 棧的壓入、彈出序列(Java)

《劍指offer》系列 棧的壓入、彈出序列(Java)

連結

牛客:棧的壓入、彈出序列

題目描述

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

思路

我們需要一個棧來模擬操作,先不停地將陣列pushA的值壓入,每壓入一個數值就需要和popA陣列對比來判斷這個值是否是彈出值,如果是彈出值,就需要從棧中丟擲,再用新的棧頂的值和popA的值判斷是否相等,相等操作依舊,不相等就退出,等到pushA陣列的所有值都壓入棧後,相關操作也已完成,再判斷棧是否為空,為空的話就說明全部彈出,兩個序列是一致的。

程式碼

import java.util.ArrayList;
import java.util.Stack;
public class Solution {
    public boolean IsPopOrder(int [] pushA,int [] popA) {
        Stack<Integer> stack = new Stack<>();
        int index_pop = 0;
        for (int i = 0; i < pushA.length; i++) {
            stack.push(pushA[i]);
            //這裡要判斷棧不為空,否則程式編譯不過去,原因在於peek函式需要stack不為空
            while (!stack.isEmpty() && stack.peek() == popA[index_pop]) {
                stack.pop();
                index_pop++;
            }
        }
        if (stack.isEmpty())
            return true;
        else
            return false;
    }
}