首页 > 其他 > 详细

重建二叉树

时间:2020-02-10 12:19:06      阅读:58      评论:0      收藏:0      [点我收藏+]
  • 输入某二叉树的前序遍历和中序遍历的结果,请重建出该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。例如输入前序遍历序列{1,2,4,7,3,5,6,8}和中序遍历序列{4,7,2,1,5,3,8,6},则重建二叉树并返回。
public class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode(int x) {
        val = x;
    }
}
import java.util.Arrays;

public class Solution {
    public TreeNode reConstructBinaryTree(int[] pre, int[] in) {

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

        // 根
        int root = pre[0];

        //根节点
        TreeNode tree = new TreeNode(root);


        //中序中root值为分割点 左侧:左子树,右侧:右子树
        int index = 0;
        for (; index < in.length - 1; index++) {
            if (in[index] == root) {//在中序遍历中找到根节点
                break;
            }
        }

        //中序遍历中:左子树的起始位置 0
        // 中序遍历中:左子树的结束位置 index
        // copyOfRange(int[] original, int from, int to)  包括from 不包括to

        //前序中pre的首元素为根节点,
        // 第二个元素是左子树的起始位置,
        // 由于左子树个数与中序的个数是一致的,通过左子树个数相等即可推出,前序中左子树的结束位置
        tree.left = reConstructBinaryTree(Arrays.copyOfRange(pre, 1, index + 1), Arrays.copyOfRange(in, 0, index));

        //中序遍历中:右子树的起始位置 index+1
        // 中序遍历中:右子树的结束位置 in.length-1
        tree.right = reConstructBinaryTree(Arrays.copyOfRange(pre, index + 1, pre.length), Arrays.copyOfRange(in, index + 1, in.length));

        return tree;
    }
}

重建二叉树

原文:https://www.cnblogs.com/VVII/p/12290163.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!