【剑指offer中等部分1】重建二叉树 (java实现)

题目描述

输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{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;
    }
}