题目描述
输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
分析
该题就是类似由前序和中序序遍历序列,重新画出二叉树的图,或者写出后序遍历序列,考试中常常会遇到这种题,我们需要知道这种题的解题方式,首先最重要的是前序遍历第一个元素即为根节点,或者后序遍历序列就是最后一个元素是根节点。由根节点元素,代入中序遍历序列,确定两边的左子树和右子树。
Arrays.copyOfRange方法
将一个原始的数组original,从下标from开始复制,复制到上标to,生成一个新的数组。
注意这里包括下标from,不包括上标to。即左闭右开,如[1,5)
代码如下:
import java.util.Arrays;
public class Solution {
//前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}
public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
if(pre.length == 0 || in.length == 0){
return null;//递归条件
}
TreeNode node = new TreeNode(pre[0]); //pre[0]即为根节点
for(int i = 0;i < pre.length;i++){
//先找到根节点对应在中序遍历中的位置
if(pre[0] == in[i]){
//i=3时,in[i]==pre[0],则中序遍历序列{4,7,2,1,5,3,8,6}可分为左边{4,7,2}和右边{5,3,8,6},继续递归
//{2,4,7}传给pre数组,{4,7,2}传给in数组
//此时pre[0]为2,i==2,pre[0]==in[2],{4,7}再传给pre数组,{4,7}也传给in数组,继续递归
//此时pre[0]为4,i==0,明显in[i]=4,in为{4,7},4作为根节点左子树为空,只有一个右子树,node.left=null,这个null是pre=null,in=null而返回得到的,node.right=7
//然后一个个回调,形成二叉树,最后最上根节点就是node
node.left = reConstructBinaryTree(Arrays.copyOfRange(pre,1,i+1), Arrays.copyOfRange(in,0,i));
node.right = reConstructBinaryTree(Arrays.copyOfRange(pre,i+1,pre.length), Arrays.copyOfRange(in,i+1,in.length));
}
}
return node;
}
}