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

题目链接

法一

    /**
     * 根据前序序列和中序序列构建二叉树
     *
     * @param preLeft  前序序列左子树/右子树范围的左边界
     * @param preRight 前序序列左子树/右子树范围的右边界
     * @param inLeft   中序序列左子树/右子树范围的左边界
     * @param inRight  中序序列左子树/右子树范围的右边界
     * @param preorder 前序序列
     * @param inorder  中序序列
     * @return
     */
    private TreeNode preorderTraversal(int preLeft, int preRight, int inLeft, int inRight, int[] preorder, int[] inorder) {
        if (preLeft > preRight || inLeft > inRight) {
            return null;
        }
        TreeNode root = new TreeNode(preorder[preLeft]); // 前序中首位肯定是根结点
        int inRoot = inLeft;
        while (inRoot <= inRight && inorder[inRoot] != preorder[preLeft]) { // 找到中序中根结点的位置
            inRoot++;
        }
        int left = inRoot - inLeft; // 可以算出左子树结点个数
        root.left = preorderTraversal(preLeft + 1, preLeft + left, inLeft, inRoot - 1, preorder, inorder);
        root.right = preorderTraversal(preLeft + left + 1, preRight, inRoot + 1, inRight, preorder, inorder);
        return root;
    }

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

本地测试

        /**
         * 105. 从前序与中序遍历序列构造二叉树
         */
        lay.showTitle(105);
        Solution105 sol105 = new Solution105();
        int[] preorder105 = new int[]{3, 9, 20, 15, 7};
        int[] inorder105 = new int[]{9, 3, 15, 20, 7};
        System.out.println(Arrays.toString(preorder105));
        System.out.println(Arrays.toString(inorder105));
        TreeNode tree105 = sol105.buildTree(preorder105, inorder105);
        treeOpt.layerOrder(tree105);