劍指Offer學習-面試題7:重建二叉樹

	/**
     * 重建二叉樹
     * 輸入某二叉樹的前序遍歷和中序遍歷的結果,請重建該二叉樹(不含重複的數字)
     *
     * @param preorder
     * @param inorder
     * @return
     */
    public TreeNode construct(int[] preorder, int[] inorder) {
        Map<Integer, Integer> map = new HashMap<>(16);
        for (int i = 0; i < inorder.length; i++) {
            map.put(inorder[i], i);
        }
        return buildTreeNode(0, inorder.length - 1, map, preorder);

    }

    private TreeNode buildTreeNode(int left, int right, Map<Integer, Integer> map, int[] preorder) {
        if (left > right) return null;
        TreeNode root = new TreeNode(preorder[left++]);
        Integer index = map.get(root.val);
        root.left = buildTreeNode(left, index - 1, map, preorder);
        root.right = buildTreeNode(index + 1, right, map, preorder);
        return root;
    }
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章