1.题目
输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。如果是则输出Yes,否则输出No。假设输入的数组的任意两个数字都互不相同。
例如:输入{5, 7, 6, 9, 11, 10, 8},则返回true, 因为这个整数序列可以找到一个二叉搜索树并且该数组序列是其后续遍历结果。如果输入的数组是{7,4,6,5},则由于没有哪颗二叉搜索树的后序遍历结果是这个序列,因此返回false。
2.思路
后序遍历得到的序列中,最后一个数字是树的根节点的值,数组中前面的数字可以分为两部分,第一部分是左子树节点的值,它们都比根节点的值小,第二部分是右子树节点的值,它们都比根节点的值大。例如题目所给数组中,最后一个数字8为根节点的值,前面,5,7,6都比8小,所以是左子树的值,后半部分9,11,10都比8大,是右子树的值。然后同样的方法再递归判断数组{5, 7,6}和数组{9,11,10}.
3.代码
# -*- coding:utf-8 -*-
class Solution:
def VerifySquenceOfBST(self, sequence):
# write code here
if len(sequence) == 0:
return False
return self.subVerifySquenceOfBST(sequence)
def subVerifySquenceOfBST(self, sequence):
if len(sequence) <= 2:
return True
flag = sequence[-1]
index = 0
while sequence[index] < flag:
index += 1
j = index
while j < len(sequence)-1:
if sequence[j] > flag:
j += 1
else:
return False
return self.subVerifySquenceOfBST(sequence[:index]) and self.subVerifySquenceOfBST(sequence[index:-1])