重建二叉树

最新推荐文章于 2025-03-31 21:54:50 发布

吾同木

最新推荐文章于 2025-03-31 21:54:50 发布

阅读量107

点赞数

CC 4.0 BY-SA版权

分类专栏： java 算法

本文链接：https://blog.youkuaiyun.com/Running_man_1001/article/details/116333661

java 同时被 2 个专栏收录

50 篇文章

订阅专栏

算法

42 篇文章

订阅专栏

本文介绍了一种根据二叉树的前序遍历和中序遍历结果来重建二叉树的方法。通过递归算法找到根节点，并以此为基础继续构建左右子树，最终完成整棵树的构建。

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

https://blog.youkuaiyun.com/u013834525/article/details/80421684

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

public TreeNode reConstructBinaryTree(int [] pre,int [] in) {
        //{4,7,2,1,5,3,8,6}中序遍历
        //{1,2,4,7,3,5,6,8}前序遍历
        /**
         * 前序 中左右
         * 中序 左中右
         * 后续 左右中
         */
        if(pre == null || in == null || pre.length == 0|| in.length == 0 || pre.length != in.length){
            return null;
        }
        return getRootNode(pre,0,pre.length-1,in,0,in.length-1);
    }

    private TreeNode getRootNode(int pre[] , int preFrom , int preEnd, int in[], int inFrom, int inEnd){
        if(preEnd < preFrom || inFrom > inEnd){
            return null;
        }

        TreeNode root  = new TreeNode(pre[preFrom]);
        int inIndex = inFrom;
        for(int i=inFrom;i<=inEnd;i++){
            if(in[i] == pre[preFrom]){
                inIndex = i;
                root.left = getRootNode(pre,preFrom+1,preFrom+(inIndex-inFrom),in,inFrom,inIndex-1);
                root.right = getRootNode(pre,preFrom+1+(inIndex-inFrom),preEnd,in,inIndex+1,inEnd);

            }
        }

        return root;
    }
    
//Definition for binary tree
class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
    @Override
    public String toString(){
        String result = "" + val;
        if(left != null && right != null){
            result = left + "," + val + "," + result;
        } else if(right != null){
            return val + "," + right;
        } else if (left != null){
            return left + "," + val;
        }else {
            return val + "";
        }
        return result;
    }
}