LeetCode 105从前序与中序遍历序列构造二叉树 详细解读

题目:给定一颗树前序遍历与中序遍历。构造二叉树并返回根节点。

示例1:

    3
   / \
  9   20
     /  \
    15   7
Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]

思路:
本质上思路就是平时考试试卷上给你前序遍历,中序遍历,还原二叉树(不是废话,让你用笔写,你肯定会,思路就是用笔写出答案的思路)。首先根据前序遍历确定根节点,之后找到根节点在中序遍历的位置,左侧的就都是左子树,右侧的就都是右子树。确定了左子树、右子树的元素个数,使用递归完成题目。

代码:

class Solution {
    
    
    private Map<Integer, Integer> helpMap;
    public TreeNode makeTree(int[] preorder, int[] inorder, int pre_left, int pre_right, int in_left, int in_right) {
    
    
        //边界条件
        if (pre_left > pre_right||in_left>in_right) {
    
    
            return null;
        }
        // 前序遍历中的第一个节点是根节点
        int pre_root = pre_left;
        // 在中序遍历中找到根节点
        int in_root = helpMap.get(preorder[pre_root]);
        
        // 先把根节点建立出来
        TreeNode root = new TreeNode(preorder[pre_root]);
        // 得到左子树中的节点数目
        int left_sum = in_root - in_left;
        // 递归地构造左子树,并连接到根节点
        // 先序遍历中从左边界+1 开始的 left_sum个元素就对应了中序遍历中从左边界到根节点-1的元素
        root.left = makeTree(preorder, inorder, pre_left + 1, pre_left + left_sum, in_left, in_root - 1);
        // 递归地构造右子树,并连接到根节点
        // 先序遍历中从左边界+左子树节点数目+1开始到右边界的元素就对应了中序遍历中从根节点+1到右边界的元素
        root.right = makeTree(preorder, inorder, pre_left + left_sum + 1, pre_right, in_root + 1, in_right);
        return root;
    }

    public TreeNode buildTree(int[] preorder, int[] inorder) {
    
    
        int n = preorder.length;
        helpMap = new HashMap<Integer, Integer>();
        for (int i = 0; i < n; i++) {
    
    
     	    //通过值来找索引,所以值为键
            helpMap.put(inorder[i], i);
        }
        return makeTree(preorder, inorder, 0, n - 1, 0, n - 1);
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_43744992/article/details/121558280