《劍指offer》系列 棧的壓入、彈出序列(Java)
阿新 • • 發佈:2018-11-25
連結
牛客:棧的壓入、彈出序列
題目描述
輸入兩個整數序列,第一個序列表示棧的壓入順序,請判斷第二個序列是否可能為該棧的彈出順序。假設壓入棧的所有數字均不相等。例如序列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; } }