劍指offer 31——棧的壓入、彈出序列

本題主要在於考察棧的特性,優化時可以考慮自己實現一個棧。

原題

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

示例 1:

輸入:pushed = [1,2,3,4,5], popped = [4,5,3,2,1]
輸出:true
解釋:我們可以按以下順序執行:
push(1), push(2), push(3), push(4), pop() -> 4,
push(5), pop() -> 5, pop() -> 3, pop() -> 2, pop() -> 1

示例 2:

輸入:pushed = [1,2,3,4,5], popped = [4,3,5,1,2]
輸出:false
解釋:1 不能在 2 之前彈出。

提示:

  • 0 <= pushed.length == popped.length <= 1000

  • 0 <= pushed[i], popped[i] < 1000

  • pushed 是 popped 的排列。

原題url:https://leetcode-cn.com/problems/zhan-de-ya-ru-dan-chu-xu-lie-lcof/

解題

借用現成的Stack類

既然本題是要判斷進棧、出棧序列是否匹配,那麼我們可以直接用現成的Stack類進行模擬。

棧的特性是先入後出,因此入棧的數字,想進行出棧:

  1. 進棧之後立刻出棧

  2. 進棧之後,等待後面進棧的數字全部出棧後,再進行出棧

那模擬的時候,就可以遍歷進棧序列,先讓當前數字進棧,然後開始讓棧中數字出棧,如果滿足出棧序列,則可以繼續出棧,直到不能出棧。

讓我們來看看代碼:

class Solution {
    public boolean validateStackSequences(int[] pushed, int[] popped) {
        // 模擬棧
        Stack<Integer> stack = new Stack<>();
        // 出棧下標
        int popIndex = 0;
        // 遍歷入棧序列
        for (int i = 0; i < pushed.length; i++) {
            // 先讓當前數字入棧
            stack.push(pushed[i]);
            // 遍歷棧
            while (!stack.isEmpty() && stack.peek() == popped[popIndex]) {
                popIndex++;
                stack.pop();
            }
        }

        return popIndex == pushed.length;
    }
}

提交OK。

複雜度分析:時間複雜度 O(N):其中 N 爲列表 pushed 的長度;每個元素最多入棧與出棧一次,即最多共 2N 次出入棧操作。忽略係數後,得出 O(N)。空間複雜度 O(N):輔助棧 stack 最多同時存儲 N 個元素。

自己實現一個簡單的棧結構

上面我們使用了 Java 中現成的類 Stack,但因爲我們這裏的場景十分簡單,而 Stack 會考慮到擴容、併發修改的情況,所以相應會對性能有一定的影響。因此我們完全可以利用數組設計一個符合本題的、簡單的棧結構。

其實代碼和上面類似,但因爲是直接使用數組實現的,因此在大數量下,性能上可以快很多。

public boolean validateStackSequences(int[] pushed, int[] popped) {
    // 利用數組實現一個棧
    int stack[] = new int[pushed.length];
    // 棧中的元素個數
    int stackSize = 0;
    // 出棧序列被遍歷到的下標
    int poppedIndex = 0;
    // 遍歷入棧序列
    for (int i = 0; i < pushed.length; i++) {
        stack[stackSize] = pushed[i];
        stackSize++;
        // 棧內有數據,並且棧頂元素等於當前出棧序列中的數字
        while (stackSize != 0 && stack[stackSize - 1] == popped[poppedIndex]) {
            // 出棧
            stackSize--;
            // 繼續比較下一個
            poppedIndex++;
        }
    }
    return poppedIndex == popped.length;
}

提交OK,複雜度和上面完全相同。

當然,這道題的難度還有可以提高的地方,原題中有假設壓入棧的所有數字均不相等,如果允許數字重複的話,你可以想到要怎麼解決嗎?

總結

以上就是這道題目我的解答過程了,不知道大家是否理解了。本題主要在於考察棧的特性,優化時可以考慮自己實現一個棧。

有興趣的話可以訪問我的博客或者關注我的公衆號,說不定會有意外的驚喜。

https://death00.github.io/

公衆號:健程之道

點擊此處留言

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章