LeetCode-106. 从中序与后序遍历序列构造二叉树-Java-medium

题目链接

法一

    private Map<Integer, Integer> mp = new HashMap<>();

    /**
     * 根据后序序列和中序序列构建二叉树
     *
     * @param inLeft    中序序列左子树/右子树范围的左边界
     * @param inRight   中序序列左子树/右子树范围的右边界
     * @param postLeft  后序序列左子树/右子树范围的左边界
     * @param postRight 后序序列左子树/右子树范围的右边界
     * @param inorder   中序序列
     * @param postorder 后序序列
     * @return
     */
    private TreeNode preorderTraversal(int inLeft, int inRight, int postLeft, int postRight, int[] inorder, int[] postorder) {
        if (inLeft > inRight || postLeft > postRight) {
            return null;
        }
        TreeNode root = new TreeNode(postorder[postRight]); // 后序中末位肯定是根结点
        int inRoot = mp.get(postorder[postRight]); // 得到中序中根节点的位置
        int left = inRoot - inLeft; // 可以算出左子树结点个数
        root.left = preorderTraversal(inLeft, inRoot - 1, postLeft, postLeft + left - 1, inorder, postorder);
        root.right = preorderTraversal(inRoot + 1, inRight, postLeft + left, postRight - 1, inorder, postorder);
        return root;
    }

    /**
     * 前序/后序+中序可以唯一确定一棵二叉树，所以可以用来建树
     * 后序：左子树 - 右子树 - 根
     * 中序：左子树 - 根 - 右子树
     * （1）后序中末位肯定是根结点root，我们可以据此找到中序中根结点的位置
     * （2）中序中根结点左边就是左子树结点，右边就是右子树结点，于是可以算出左子树结点个数
     * （3）根据左子树结点个数，可以确定后序中左子树结点和右子树结点的范围
     * （4）左子树：root.left = preorderTraversal(中序左子树范围，后序左子树范围，中序序列，后序序列)
     * （5）右子树：root.right = preorderTraversal(中序右子树范围，后序右子树范围，中序序列，后序序列)
     *
     * @param inorder
     * @param postorder
     * @return
     */
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        for (int index = 0; index < inorder.length; index++) {
            mp.put(inorder[index], index); // 构造哈希映射，帮助我们快速定位中序中根节点的位置
        }
        return preorderTraversal(0, inorder.length - 1, 0, postorder.length - 1, inorder, postorder);
    }

本地测试

        /**
         * 106. 从中序与后序遍历序列构造二叉树
         */
        lay.showTitle(106);
        Solution106 sol106 = new Solution106();
        int[] inorder106 = new int[]{9, 3, 15, 20, 7};
        int[] postorder106 = new int[]{9, 15, 7, 20, 3};
        System.out.println(Arrays.toString(inorder106));
        System.out.println(Arrays.toString(postorder106));
        TreeNode tree106 = sol106.buildTree(inorder106, postorder106);
        treeOpt.layerOrder(tree106);