题目描述:
输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。如果是则输出Yes,否则输出No。假设输入的数组的任意两个数字都互不相同。
思路:
已知条件:后序序列最后一个值为root;二叉搜索树左子树值都比root小,右子树值都比root大。
1、确定root;
2、遍历序列(除去root结点),找到第一个大于root的位置,则该位置左边为左子树,右边为右子树;
3、遍历右子树,若发现有小于root的值,则直接返回false;
4、分别判断左子树和右子树是否仍是二叉搜索树(即递归步骤1、2、3)。
代码:
# -*- coding:utf-8 -*-
class Solution:
def __init__(self):
self.time = 0
def VerifySquenceOfBST(self, sequence):
# write code here
if self.time == 0 and len(sequence) == 0:
return False
self.time += 1
if len(sequence) < 2:
return True
j = -1
for i in range(len(sequence)-1):
if j == -1:
if sequence[i] > sequence[len(sequence)-1]:
j = i
else:
if sequence[i] < sequence[len(sequence)-1]:
return False
return self.VerifySquenceOfBST(sequence[:j]) and self.VerifySquenceOfBST(sequence[j:len(sequence)-1])