劍指offer:二叉搜索樹的後序遍歷序列

題目描述
輸入一個整數數組,判斷該數組是不是某二叉搜索樹的後序遍歷的結果。如果是則輸出Yes,否則輸出No。假設輸入的數組的任意兩個數字都互不相同。

class Solution:
    """
    一個二叉搜索樹BST滿足: max(左子樹) < 根節點 < min(右子樹)
    由於題目給出的是一個後序遍歷,那麼序列的最後一個元素就應該是根節點。
    因此我們從BST的定義出發,遍歷整個序列,找到第一個大於根節點的元素k,k以前的元素屬於左子樹,
    從k開始到根節點之前的元素屬於右子樹。
    如果遍歷整個序列之後得到的左右子樹都滿足BST的定義,則遞歸判斷左子樹和右子樹是否滿足BST定義
    """
    def VerifySquenceOfBST(self, sequence):
        if not sequence:
            return False

        root = sequence[-1]  # 獲取根節點
        """
        查找第一個大於根節點的元素,得到左右子樹的分界點
        """
        idx = 0
        while idx < len(sequence) - 1:
            if sequence[idx] > root:
                break
            idx += 1

        # 需要驗證右子樹是否滿足BST,即所有右子樹的節點都大於根節點,
        # 如果不滿足則這個序列不是BST的後序遍歷
        for i in range(idx, len(sequence) - 1):
            if sequence[i] < root:
                return False

        # 如果這個序列是BST的後序遍歷,那麼遞歸判斷左右子樹是否分別滿足BST
        left = True
        if idx > 0:
            left = self.VerifySquenceOfBST(sequence[: idx])

        right = True
        if idx < len(sequence) - 1:
            right = self.VerifySquenceOfBST(sequence[idx: -1])

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