从前序与中序遍历序列构造二叉树(来源力扣)

最新推荐文章于 2024-09-27 15:30:00 发布

「已注销」

最新推荐文章于 2024-09-27 15:30:00 发布

阅读量139

点赞数

文章标签： leetcode 深度优先算法

本文链接：https://blog.csdn.net/qq_53198408/article/details/123292021

版权

二叉树先序遍历中序遍历构造算法递归

关键词由CSDN通过智能技术生成

给定两个整数数组 preorder 和 inorder ，其中 preorder 是二叉树的先序遍历， inorder 是同一棵树的中序遍历，请构造二叉树并返回其根节点。

输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
输出: [3,9,20,null,null,15,7]

1 <= preorder.length <= 3000
inorder.length == preorder.length
-3000 <= preorder[i], inorder[i] <= 3000
preorder 和 inorder 均 无重复 元素
inorder 均出现在 preorder
preorder 保证为二叉树的前序遍历序列
inorder 保证为二叉树的中序遍历序列

class Solution {
	//使用map可以快速定位到inorder中值的索引
    Map<Integer, Integer> map = new HashMap<>();
    public TreeNode buildTree(int[] preorder, int[] inorder) {
        int n = preorder.length;
        for(int i = 0; i < n; i++)
            map.put(inorder[i], i);
        return help(preorder, inorder, 0, n - 1, 0, n - 1);
    }
    private TreeNode help(int[] preorder, int[] inorder, int preLeft, int preRight, int inLeft, int inRight) {
        if(preLeft > preRight) return null;
        //找到前序遍历第一个节点在inorder中的索引（也就是根节点）
        int inRootIdx = map.get(preorder[preLeft]);
        //根据根节点构建root
        TreeNode root = new TreeNode(inorder[inRootIdx]);
        //计算左子树的长度
        int subtreeLength = inRootIdx - inLeft;
        //递归求左子树
        root.left = help(preorder, inorder, preLeft + 1, preLeft + subtreeLength, inLeft, inRootIdx - 1);
        //递归求右子树
        root.right = help(preorder, inorder, preLeft + subtreeLength + 1, preRight, inRootIdx + 1, inRootIdx);
        return root;
    }
}