代码随想录之根据前序与中序序列构造二叉树-优快云博客

该文提供了两种方法，使用JAVA实现从给定的前序和中序序列以及后序和中序序列构建二叉树的算法。主要思路是利用哈希映射存储中序序列的索引，然后通过递归找到子树的根节点并构建左右子树。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

从前序与中序序列构造二叉树力扣105
从后续与中序序列构造二叉树力扣106
本题思路学习于代码随想录

JAVA版本

从前序与中序序列构造二叉树

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    Map<Integer,Integer> map;
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        map = new HashMap<>();
        for(int i =0; i<inorder.length;i++){
            map.put(inorder[i] , i) ;          //将中序的数组放入map中，值为key，对应的i为索引。
        }

        return findNode(preorder, 0, preorder.length, inorder,  0, inorder.length);
    }
    TreeNode findNode(int[] preorder, int preBegin, int preEnd, int[] inorder, int inBegin, int inEnd){
        // 参数里的范围都是前闭后开
        if (preBegin >= preEnd || inBegin >= inEnd) {  // 不满足左闭右开，说明没有元素，返回空树
            return null;
        }
        int rootIndex = map.get(preorder[preBegin]);  // 找到前序遍历的第一个元素在中序遍历中的位置
        TreeNode root = new TreeNode(inorder[rootIndex]);  // 构造结点
        int lenOfLeft = rootIndex - inBegin;  // 保存中序左子树个数，用来确定前序数列的个数
        root.left = findNode(preorder, preBegin + 1, preBegin + lenOfLeft + 1,
                            inorder, inBegin, rootIndex);
        root.right = findNode(preorder, preBegin + lenOfLeft + 1, preEnd,
                            inorder, rootIndex + 1, inEnd);

        return root;
    }
}

从后序与中序序列构造二叉树

class Solution {
    Map<Integer,Integer> map;
    public TreeNode buildTree(int[] inorder, int[] postorder) {
        map = new HashMap<>();
        for(int i =0; i<inorder.length;i++){
            map.put(inorder[i] , i) ;          //将中序的数组放入map中，值为key，对应的i为索引。
        }

        return findNode(inorder,  0, inorder.length, postorder,0, postorder.length);
    }
    TreeNode findNode(int[] inorder, int inBegin, int inEnd, int[] postorder, int postBegin, int postEnd){
            // 参数里的范围都是前闭后开,前闭后开是因为是数组，数组从0开始。
        if (inBegin >= inEnd || postBegin >= postEnd) {  // 不满足左闭右开，说明没有元素，返回空树
            return null;
        }
        int rootIndex = map.get(postorder[postEnd - 1]);  // 找到后序遍历的最后一个元素在中序遍历中的位置
        TreeNode root = new TreeNode(inorder[rootIndex]);  // 构造结点
        int lenOfLeft = rootIndex - inBegin;  // 保存中序左子树个数，用来确定后序数列的个数
        root.left = findNode(inorder, inBegin, rootIndex, postorder, postBegin, postBegin + lenOfLeft);
        root.right = findNode(inorder, rootIndex + 1, inEnd, postorder, postBegin + lenOfLeft, postEnd - 1); //+1是因为左闭右开，所有取的是+1

        return root;
    }
}