剑指offer-33 二叉搜索树的后序遍历序列
微信搜索【程序员画工师】关注更多Java编程技术、数据结构与算法、面试题相关内容。
题目
输入一个整数数组,判断该数组是不是某二叉搜索树的后序遍历的结果。假设输入的数组的任意两个数字都互不相同。
思路
以{5,7,6,9,11,10,8}为例,后序遍历结果的最后一个数字8就是根结点的值。在这个数组中,前3个数字5、7和6都比8小,是值为8的结点的左子树结点;后3个数字9、11和10都比8大,是值为8的结点的右子树结点。
所以我们可以得到规律:在后序遍历得到的序列中,最后一个数字是树的根结点的值。数组中前面的数字可以分为两部分:第一部分是左子树结点的值,它们都比根结点的值小;第二部分是右子树结点的值,它们都比根结点的值大。
因此,我们可以总结出算法步骤:
Step1.通过取出序列最后一个元素得到二叉搜索树的根节点;
Step2.在二叉搜索树中左子树的结点小于根结点,因此可以遍历一次得到比根节点值小的节点集合即为左子树;
Step3.在二叉搜索树中右子树的结点大于根结点,因此可以继续遍历后序元素得到右子树,在遍历右子树的过程中判断右子树中的节点是否符合都比根节点的值大的要求;
Step4.重复以上步骤递归判断左右子树是不是二叉搜索树,如果都是,则返回true,如果不是,则返回false;
上代码
public class Solution {
public boolean VerifySquenceOfBST(int [] sequence) {
if(sequence.length == 0){
return false;
}
return verify(sequence,0,sequence.length-1);
}
public boolean verify(int [] sequence,int startIndex,int endIndex){
if(endIndex - startIndex<= 1){
return true;
}
int leftIndex = startIndex;
int root = sequence[endIndex];
//先找到左子树的位置
while(leftIndex < endIndex && sequence[leftIndex] <= root){
leftIndex ++;
}
//判断右子树中的节点是否符合二叉搜索树的规则
int rightIndex = leftIndex;
for(;rightIndex<endIndex;rightIndex++){
if(sequence[rightIndex] < root){
return false;
}
}
//递归处理左右子树
return verify(sequence,startIndex,leftIndex-1) && verify(sequence,rightIndex,endIndex-1);
}
}
举一反三
如果处理一棵二叉树的遍历序列,则可以先找到二叉树的根节点,再基于根节点把整棵树的遍历序列拆分为左右子树分别对应的子序列,然后递归处理。题7 《重建二叉树》也是这个思路。
References
[1] 《剑指offer(第二版)》 何海涛著
程序员画工师公众号,获取更多详细Java、Python、C、前端、小程序、产品相关学习资料,欢迎交流