题目描述
输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。如果是则输出Yes,否则输出No。假设输入的数组的任意两个数字都互不相同。
题目链接
本题的知识点:二叉搜索树,后序遍历
解题思路
由于后序遍历中最后一个节点为根节点,我们先取出根节点,然后根据二叉搜索树的性质知道左子树的node上的任何值都小于root的值。可以找到root的左子树数组,然后认为剩下的部分为右子树,判断其是否满足我二叉搜索树右子树上的node的值都大于根节点的要求。如果不满足则返回False,否则对子数组进行递归运算。
将数组分成左右两个部分,即为该根节点的左右子树。
本题需要注意的地方:
不能直接使用return self.VerifySquenceOfBST(sequence[:index]) and self.VerifySquenceOfBST(sequence[index:-1])
进行递归,因为初始的空数组返回的False,而递归后的空数组返回的是True,所以通过一个判断避免对空数组进行递归。
class Solution:
def VerifySquenceOfBST(self, sequence):
# write code here
length = len(sequence)
if length <= 0:
return False
root,index= sequence[-1],length-1
for i,node in enumerate(sequence[:-1]):
if node > root:
index = i
break
for node in sequence[index:-1]:
if node <= root:
return False
left,right = True,True
if index >0: left = self.VerifySquenceOfBST(sequence[:index])
if index < length -1: right = self.VerifySquenceOfBST(sequence[index:-1])
return left and right