Leetcode刷题Python剑指 Offer 33. 二叉搜索树的后序遍历序列

Posted 2022-11-23 Better Bench

1 题目

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

参考以下这颗二叉搜索树：

     5
    / \\
   2   6
  / \\
 1   3

示例 1：

输入: [1,6,3,2,5]
输出: false
示例 2：

输入: [1,3,2,6,5]
输出: true

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/er-cha-sou-suo-shu-de-hou-xu-bian-li-xu-lie-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2 解析

（1）先找到根节点的右子节点
根节点是后续遍历序列的最后一个值，从后向前遍历序列，找到比根节点小的第一个数就是根节点的右子节点

（2）然后判断右子树的值是否全大于root，左子树的值是否都小于root

（3）然后再递归根节点的左右子树即可

3 python实现

class Solution:
    def verifyPostorder(self, postorder: List[int]) -> bool:

        if not postorder:return True
        root = postorder[-1]
        idx = 0
        for i in range(len(postorder)):
            if postorder[i]>=root:
                idx = i
                break
        left = postorder[:idx]
        right  = postorder[idx:-1]
        # 判断右子树的节点都大于根节点
        for val in right:
            if val<root:
                return False
        # 判断左子树的节点都小于根节点
        for val in left:
            if val>root:
                return False
        return self.verifyPostorder(left) and self.verifyPostorder(right)