判断数组是不是某二叉搜索树的后序遍历的结果
参考: 题目描述 输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。如果是则输出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
下一篇:
数据结构->二叉树的基本操作