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