剑指offer重建二叉树算法题Java实现（二叉树，递归）

题目描述

题目描述
输入某二叉树的前序遍历和中序遍历的结果，请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6}，则重建二叉树并返回。

问题分析

二叉树大家都很熟悉，先声明一下二叉树的结构。

Definition for binary tree
   public class TreeNode {
      int val;
       TreeNode left;
       TreeNode right;
       TreeNode(int x) { 
       		val = x; 
       }
    }

要解决这个问题，我们首先要知道什么是二叉树的前序/中序遍历。下面以打印当前节点的值为例
前序遍历指的是：
先输出当前节点的值，然后对当前节点的左子节点进行前序遍历，再对当前节点的右子节点进行前序遍历。
中序遍历指的是：
对当前节点的左子节点进行中序遍历，然后输出当前结点的值，再对当前节点的右子节点进行中序遍历。

代码及讲解

数组pre[]表示前序遍历的节点集合本题中为{1,2,4,7,3,5,6,8}
数组in[]表示中序遍历的集合本题中为{4,7,2,1,5,3,8,6}
整型len表示节点的个数。
1.前序遍历，则pre[0]一定是根节点root。在in[]中寻找根节点的下标n，根据中序遍历的性质，则下标小于n的都是根节点的左子树上的节点，大于n的则为右子树。
2.利用Arrays.copyOfRange(int [] arr,int from,int to)方法截取数组（对arr进行操作，从 from到to的元素复制进一个新的数组并返回，元素包括from但是不包括to）。
2.1对于pre来说，截取从pre[1]-pre[n+1]的元素，不包括pre[n+1]。对in来说，截取in[0]-in[n]的元素，不包括pre[n]。截取后的pre和in分别是原根节点的左子树的前序遍历和中序遍历顺序集合。
2.2同样的，对于pre来说，截取从pre[n+1]-pre[len]的元素。对in来说，截取in[n+1]-in[len]的元素。截取后的pre和in分别是原根节点的右子树的前序遍历和中序遍历顺序集合。
3.进行判断
如果pre.length仅有一个元素，则直接返回，如果pre.length=0，则返回null，表示当前根节点的左/右子树为空。
4.根据以上的条件进行递归处理。代码如下。

import java.util.*;
public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        if(pre.length==0||in.length==0)return null;
        TreeNode root=new TreeNode(pre[0]);
        if(pre.length==1)return root;
        int n=find(in,pre[0]);
        root.left=reConstructBinaryTree(Arrays.copyOfRange(pre,1,n+1),Arrays.copyOfRange(in,0,n));
        root.right=reConstructBinaryTree(Arrays.copyOfRange(pre,n+1,pre.length),Arrays.copyOfRange(in,n+1,in.length));
        return root;
    }
    public int find(int[] arr,int num){
        int res=0;
        for(int i=0;i<arr.length;i++){
            if(arr[i]==num){
                res=i;
                break;
            }
        }
        return res;
    }
 
}

总结

这道题是我第二次做了，第一次完全没有头绪，第二次知道怎样根据遍历顺序构建二叉树，但是在截取数组那里卡了壳，最后靠着度娘才有了些眉目，发现自己的水平还是差得远，各位一起加油，长风破浪会有时。