剑指 Offer 33. 二叉搜索树的后序遍历序列(python)

题目链接

题目描述：

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

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

示例 1：

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

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

解题思路：
1.后序遍历最后一个数字即为根节点
2.二叉搜索树，左子树全部小于根节点，右子树全部大于根节点
3.找到第一个大于root的节点，如果其后面节点均大于root，符合条件，递归判断左子树和右子树，否则返回false

class Solution:
    def verifyPostorder(self, postorder: List[int]) -> bool:
        #print(postorder)
        if len(postorder)==1 or len(postorder)==0:
            return True
        root=postorder[-1]
        left,right=[],[]
        for i in range(len(postorder)-1):
            node=postorder[i]
            if node<root:
                left.append(node)
            elif node==root:
                return False
            else:
                #print(i,postorder[i:-1])
                if min(postorder[i:-1])<=root:
                    return False
                else:
                    right=postorder[i:-1]
                    break
        #print(left,right)
        return self.verifyPostorder(left) and self.verifyPostorder(right)

class Solution:
    def verifyPostorder(self, postorder: List[int]) -> bool:
        
        def helper(nums):
            if len(nums) <= 1: return True
            root = nums[-1]
            for i in range(len(nums)):
                if nums[i] > root: break
            for j in range(i, len(nums)-1):
                if nums[j] < root: return False
            return helper(nums[:i]) and helper(nums[i:-1])
        
        if not postorder: return True
        return helper(postorder)