判断数组是不是某二叉搜索树的后序遍历的结果

参考: 题目描述 输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。如果是则输出Yes,否则输出No。假设输入的数组的任意两个数字都互不相同。

分析: 已知条件:后序序列最后一个值为root;二叉搜索树左子树值都比root小,右子树值都比root大。 1、确定root; 2、遍历序列(除去root结点),找到第一个大于root的位置,则该位置左边为左子树,右边为右子树; 3、遍历右子树,若发现有小于root的值,则直接返回false; 4、分别判断左子树和右子树是否仍是二叉搜索树(即递归步骤1、2、3)。

思路很简单,主要是考虑代码健壮性

# -*- coding:utf-8 -*-

class Solution:
    def VerifySquenceOfBST(self, sequence):
        if len(sequence) == 0 or sequence==None:
            return False
        root = sequence.pop(-1)

        #找到左右子树的序列值,分割点
        index = -1
        for i, v in enumerate(sequence):
            if v > root:
                index = i
                break
        if index == -1:
            return True

        left = sequence[:index]
        right = sequence[index:]

        if min(right) < root:
            return False
        
        l = True
        r = True
        if len(left) > 0:
            l = self.VerifySquenceOfBST(left)
        
        if len(right) > 0:
            r = self.VerifySquenceOfBST(right)
        return l and r
经验分享 程序员 微信小程序 职场和发展