题目描述:
输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。如果是则输出Yes,否则输出No。假设输入的数组的任意两个数字都互不相同。
思想:首先我们需要明确序列是后序遍历,因此最后一个节点一定是根节点。如果满足二叉搜索树,前部分的序列会分成两部分,前半部分是左子树并且值小于根节点,后半部分是右子树并且值大于根节点。整个过程分三个步骤:
1.遍历数组,找到区分左右子树的节点
2.判断右子树是否满足其值都大于根节点的要求
3.分离左右子树再分别判断
整个过程的核心就是分治的思想。
# -*- coding:utf-8 -*-
class Solution:
def VerifySquenceOfBST(self, sequence):
# write code here
k=len(sequence)
if k==0:#判断第一次列表是否为空
return False
if k<=2:#只要小于等于2说明满足搜索二叉树
return True
root=sequence[-1]#获取根节点
for i in range(k):#找到区分左右子树的节点
if sequence[i]>root:
break
for j in range(i,k):#如果右子树比根节点小,说明不满足
if sequence[j]<root:
return False
left=sequence[:i]#左子树
right=sequence[i:-1]#右子树
if len(left)==0:
return self.VerifySquenceOfBST(right)
if len(right)==0:
return self.VerifySquenceOfBST(left)
return self.VerifySquenceOfBST(left) and self.VerifySquenceOfBST(right)
if __name__=="__main__":
a=Solution()
s=[4,8,6,12,16,14,10]#测试用例1
k=[1,2,3,4,5]#测试用例2
print a.VerifySquenceOfBST(k)