Leetcode 二叉搜索树的后序遍历序列

leetcode

问题

输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历结果。如果是则返回 true,否则返回 false。假设输入的数组的任意两个数字都互不相同。

思路

二叉搜索树: 左子树的值 < 根节点的值 < 右子树的值
后序遍历为: 左子树 , 右子树, 根节点
那么给出数组的最后一位为root
所有值小于根节点的都为左子树,其余为右子树
用递归算法,可以接着判断左右子树是否为二叉搜索树

class Solution:
    def verifyPostorder(self, postorder: List[int]) -> bool:
        if not postorder or len(postorder)==1:
            return True
        root = postorder[-1]
        index = 0
        for i in range(len(postorder)):
            if postorder[i]>=root:
                index = i
                break

        left = postorder[:index]
        right = postorder[index:-1]
        for num in right:
            if num < root:
                return False
        return self.verifyPostorder(left) and self.verifyPostorder(right)

你可能感兴趣的:(刷题)