牛客解题思路：重建二叉树

最新推荐文章于 2022-07-23 13:01:43 发布

小姬炖蘑菇o0

最新推荐文章于 2022-07-23 13:01:43 发布

阅读量144

点赞数 1

本文链接：https://blog.csdn.net/qq_39879315/article/details/108095294

版权

重建二叉树

思路：输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
前序遍历的第一个值为根节点的值，使用这个值将中序遍历结果分成两部分，左部分为树的左子树中序遍历结果，右部分为树的右子树中序遍历的结果。然后分别对左右子树递归地求解， 注意Arrays.copyOfRange方法的使用也是包前不包后。

public TreeNode buildTree(int[] preorder, int[] inorder) {
    if(preorder.length == 0) return null;
    TreeNode root = new TreeNode(preorder[0]);
    int index = 0;
    for(int i=0;i<inorder.length;i++){
        if(inorder[i] == preorder[0]){
            index = i;
            break;
        }
    }
    root.left = buildTree(Arrays.copyOfRange(preorder, 1, 1 + index), Arrays.copyOfRange(inorder, 0, index));
    root.right = buildTree(Arrays.copyOfRange(preorder, 1 + index, preorder.length), Arrays.copyOfRange(inorder, index + 1, preorder.length));
    return root;
}

当然也可以另写一个递归函数：
通过先序遍历，找到根节点。并创建节点。 ->
然后从中序遍历中找到根的位置，左右两边就是左右子树的中序遍历。->
根据左右子树长度，可以在先序遍历中找到左右子树的先序遍历。->
递归调用。
注意：为了节约空间，传入的参数是数组，与起始位置。hashmap定位根节点时间更快。

    public TreeNode buildTree(int [] pre,int [] in) {
        TreeNode root = reConstructBTree(pre,0,pre.length-1,in,0,in.length-1);
        return root;
    }
    public TreeNode reConstructBTree(int [] pre,int startPre,int endPre,int [] in,int startIn,int endIn){
        if(startPre>endPre||startIn>endIn){
            return null;
        }
        TreeNode root = new TreeNode(pre[startPre]);
        for(int i=startIn;i<=endIn;i++){
            if(in[i]==pre[startPre]){
                root.left=reConstructBTree(pre,startPre+1,startPre+i-startIn,in,startIn,i-1);
                root.right=reConstructBTree(pre,i-startIn+startPre+1,endPre,in,i+1,endIn);
            }
        }
        return root;
    }