详解重构二叉树

重建二叉树

http://www.nowcoder.com/questionTerminal/8a19cbe657394eeaac2f6ea9b0f6fcf6

本质上就是前序遍历的框架。
前序数组的第一个是根节点,找到中序数组它的索引 (可用map降低时间复杂度),中序数组以它为中心,左边是左子树,右边是右子树。
将大问题化成小问题,利用递归。 而指针可以在每轮递归中作为通信的工具。
上一层递归告诉下一层“你该缩小范围找了”。
left每一层递归返回给上一层的是刚刚遍历完的根节点的下一个,而right返回的是刚刚遍历完的(根节点当前位置 + 左子树的节点个数)的下一个。这跟前序数组的前序遍历性质有关,而我们又是基于前序数组来重构二叉树的。

class Solution {
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        //用map来缓存值和对应的索引,方便在递归方法中查找,降低时间复杂度
        HashMap<Integer, Integer> map = new HashMap<>();
        for(int i = 0; i < inorder.length; i++){
            map.put(inorder[i] , i);
        }
        return builTree(preorder, 0, preorder.length, inorder, 0, inorder.length, map);

    }
    private TreeNode builTree(int[] preorder, int p_start, int p_end, int[] inorder, int i_start, int i_end, HashMap<Integer, Integer> map){
        //递归出口 当指针相等时,左右子树已经找完了
        if(p_start == p_end)
            return null;

        int root_val = preorder[p_start];
        //构建节点
        TreeNode root = new TreeNode(root_val);
        //找到索引,利用这个位置将数组分成2部分, 前序数组的左部分就是左子树 
        int i_root_index = map.get(root_val);
        int leftNum = i_root_index - i_start;

        //前序遍历 注意左右子树的索引 start和end区间就是一个子树
        root.left = builTree(preorder, p_start + 1, p_start + leftNum + 1, inorder, i_start, i_root_index, map);
        root.right = builTree(preorder, p_start + leftNum + 1, p_end, inorder, i_root_index + 1, i_end, map);
        return root;
    }
}
全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务