JZ7重建二叉树

描述

给定节点数为 n 的二叉树的前序遍历和中序遍历结果    
    
    
  ，请重建出该二叉树并返回它的头结点  
    
    
    
。 例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6} 提示: 1.vin.length == pre.length 2.pre 和 vin 均无重复元素 3.vin出现的元素均出现在 pre里 4.只需要返回根结点   


     


     


     
，系统会自动输出整颗树做答案对比

具体做法：

step 1：先根据前序遍历第一个点建立根节点

     


     


     


  。 step 2：然后遍历中序遍历找到根节点在数组中的位置 




 




 




 
。 step 3：再按照子树的节点数将两个遍历的序列分割成子数组



 




 




 




，将子数组送入函数建立子树  

   

   

   

。 step 4：直到子树的序列长度为0   

   

   

 ，结束递归

     

     

     

 。

代码

package mid.JZ7重建二叉树; import java.util.*; class TreeNode { int val; TreeNode left; TreeNode right; TreeNode(int x) { val = x; } } public class Solution { public TreeNode reConstructBinaryTree(int[] pre, int[] vin) { int n = pre.length; int m = vin.length; if (n == 0 || m == 0) return null; //前序遍历第一个为根节点 TreeNode root = new TreeNode(pre[0]); for (int i = 0; i < vin.length; i++) { if (pre[0] == vin[i]) { //左子树 root.left = reConstructBinaryTree(Arrays.copyOfRange(pre, 1, i + 1), Arrays.copyOfRange(vin, 0, i)); root.right = reConstructBinaryTree(Arrays.copyOfRange(pre, i + 1, pre.length), Arrays.copyOfRange(vin, i + 1, vin.length)); break; } } return root; } }