【剑指offer第四题】重建二叉树

最新推荐文章于 2019-11-29 13:42:29 发布

原创最新推荐文章于 2019-11-29 13:42:29 发布 · 174 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

21 篇文章

订阅专栏

本文深入探讨了二叉树的重建算法，通过前序和中序遍历数组，详细阐述了如何递归地构建出完整的二叉树结构。文章提供了完整的Java代码实现，为读者理解二叉树的构建过程提供了清晰的指导。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >


  //定义二叉树
  class TreeNode {
      int val;
      TreeNode left;
      TreeNode right;
      TreeNode(int x) { val = x; }
  }


//重建二叉树
public class Solution {
    public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        //建立根节点
        TreeNode treeNode=new TreeNode(pre[0]);
        
        //找到中序遍历中根节点的位置i
        int rootIndex=0;
        for(int i=0;i<in.length;i++){
            if(pre[0]==in[i]){
                rootIndex=i;
            }
        }
        //将前序遍历左右树分为两个数组
        int[] preLeft=new int[rootIndex];
        int[] preRight=new int[pre.length-rootIndex-1];
        for(int i=0;i<rootIndex;i++){
            preLeft[i]=pre[i+1];
        }
        for(int i=0;i<pre.length-rootIndex-1;i++){
            preRight[i]=pre[rootIndex+i+1];
        }
        
        //将中序遍历的树分为左右节点
        int[] midLeft=new int[rootIndex];
        int[] midRight=new int[in.length-rootIndex-1];
        for(int i=0;i<rootIndex;i++){
            midLeft[i]=in[i];
        }
        for(int i=0;i<pre.length-rootIndex-1;i++){
            midRight[i]=in[rootIndex+i+1];
        }
        
        //通过循环将重建左右树
        if(pre.length>0&in.length>0){
            if(preLeft.length>0){
                treeNode.left=reConstructBinaryTree(preLeft,midLeft);
            }
            if(preRight.length>0){
                treeNode.right=reConstructBinaryTree(preRight,midRight);
            }
        }
        return treeNode;
        
    }
}