题目介绍
输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。如果是则输出Yes,否则输出No。假设输入的数组的任意两个数字都互不相同。
具体思路
其实这个题目的思路并不难,只要知道后续遍历的思维以及二叉搜索树的特性即可轻松破解。由于是后序遍历,一般来说数组中的最后一位为根节点,而根节点的左子树部分则位于后序遍历序列的前半段,而在根节点与左子树中间的部分即为二叉搜索树的右子树。
[9, 16, 10, 20, 32, 24, 18]
即为一个常规的二叉搜索树的后序遍历序列,但是同时我们也需要考虑到一些特殊的后序遍历序列,例如[5, 4, 3, 2, 1]
,[6, 8, 12, 17, 23]
这类序列也属于二叉搜索树的后序遍历序列,一般这种都是只有一颗左子树或者只有一颗右子树的二叉搜索树。
对于上述的两类特殊的后序遍历序列可以较为直观的进行判断,而对于常规的二叉搜索树的后序遍历序列,则需要进行进一步的操作。例如序列[9, 16, 10, 20, 32, 24, 18]
,可以将其左右子树的序列分割出来,左子树[9, 16, 10]
,右子树[20 32 24]
,然后进行递归,而递归的结束条件为:
- 输入的序列类似于
[5, 4, 3, 2, 1]
或[6, 8, 12, 17, 23]
这种单调序列时,返回true
; - 如果由输入序列分割获得的右子树序列中包含小于根节点的值的元素,则返回
false
。
具体代码如下
# -*- coding:utf-8 -*-
class Solution:
def VerifySquenceOfBST(self, sequence):
# write code here
if not sequence:
return False
# 如果输入序列为单调序列,则返回True
if sequence == sorted(sequence) or sequence == list(reversed(sorted(sequence))):
return True
left = [] # 用于存放左子树序列
right = [] # 用于存放右子树序列
for i in range(len(sequence)):
if sequence[i] > sequence[-1]: # 获得左右子树序列
left += sequence[:i]
right += sequence[i:-1]
break # 一旦获得了左右子树序列便跳出循环
flag = True
for j in right: # 对右子树中的元素进行遍历,如果有小于根节点的元素即判断当前输入序列不满足条件
if j < sequence[-1]:
flag = False
if flag:
return self.VerifySquenceOfBST(left) and self.VerifySquenceOfBST(right)
else:
return False