重建二叉树(JAVA实现)

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

解题思路
前序遍历的一个结果是根节点。这样可以根据中序遍历中根节点的位置将二叉树分成两个字树(左子树、右子树),利用递归的思想可以解决这个问题。

实现代码:

/**
 * @Description:
 * @ClassName algorithm
 * @Author: Ruiwen
 * @Date: 2021/1/14 21:06
 */
public class Test005 {
    //通过传进前序遍历,和中序遍历 重建二叉树(可以输出新建的二叉树的后序遍历)
    //递归方法重构树
    public static TreeNode reConstructBinaryTree
    (int[] pre, int[] in) {
        if (pre == null || in == null) {
            return null;
        }
        return ConstructCore(pre, in,
                0, pre.length - 1, 0, in.length - 1);
    }

    //重构二叉树核心代码核心代码
    public static TreeNode ConstructCore(int[] pre, int[] in,
     int preStart, int preEnd, int inStart, int inEnd) {
        //通过前序遍历序列找到根节点,很显然是序列第一个值
        //创建根节点
        TreeNode root = new TreeNode(pre[preStart]);
        root.left = null;
        root.right = null;
        //当遇到特殊情况
        if (preStart == preEnd && inStart == inEnd) {
            return root;
        }
        //通过前序遍历序列所得到的根节点在中序遍历根节点的位置
        int rootInorder = 0;
        for (rootInorder = inStart; rootInorder <= inEnd; rootInorder++) {
            if (in[rootInorder] == pre[preStart])
                break;    
        }

        //开始分割左右子树
        //左子树长度
        int leftTreeLength = rootInorder - inStart;
        //右子树长度
        int rightTreelength = inEnd - rootInorder;
        //开始构建左右子树
        //构建左子树
        if (leftTreeLength > 0) {
            root.left = ConstructCore(pre, in,
                    preStart + 1, preStart + leftTreeLength,
                    inStart, rootInorder - 1);
        }
        //构建右子树
        if (rightTreelength > 0) {
            root.right = ConstructCore(pre, in,
                    preStart + leftTreeLength + 1, preEnd,
                    rootInorder + 1, inEnd);
        }
        return root;
    }

    public static void main(String[] args) {
        int []pre=new int[]{1,2,3,4,5,6,7};
        int []in=new int[]{3,2,4,1,6,5,7};
        System.out.println(reConstructBinaryTree(pre, in).val);
    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值