【剑指offer】重建二叉树

题目链接

题目描述

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

---

解题思路：

例如：
前序序列{1,2,4,7,3,5,6,8} = pre
中序序列{4,7,2,1,5,3,8,6} = in

• 根据前序序列得知第一个结点为根节点，根节点为1
• 找到1在中序遍历的位置，根据在中序遍历的位置进行切割，左边为左子树，右边即为右子树
• 对子树进行同样的方法进行分割

/**
* @author: hyl
* @date: 2019/08/15
**/
public class Que4 {

   public static class TreeNode {
       int val = 0;
       TreeNode left = null;
       TreeNode right = null;

       public TreeNode(int val) {
           this.val = val;

       }

   }

   //根据前序和中序构造二叉树
   public TreeNode reConstructBinaryTree(int [] pre, int [] in) {

      if (pre == null || in == null){
          return null;
      }

      if (pre.length == 0 || in.length == 0){
          return null;
      }

      if (pre.length != in.length){
          return null;
      }

      TreeNode root = new TreeNode(pre[0]);

      for (int i = 0; i < pre.length; i++) {
          if (pre[0] == in[i]){

              root.left = reConstructBinaryTree(Arrays.copyOfRange(pre , 1 , i + 1),
                       Arrays.copyOfRange(in , 0 , i));

              root.right = reConstructBinaryTree(Arrays.copyOfRange(pre , i + 1 , pre.length),
                       Arrays.copyOfRange(in , i + 1 , in.length));

           }
       }

       return root;
   }
}

代码地址：
https://github.com/Han-YLun/jianzhiOffer/blob/master/Solution/src/Que4.java

文章为阿伦原创,如果文章有错的地方欢迎指正，大家互相交流。