leetcode刷题记录（七十八）——105. 从前序与中序遍历序列构造二叉树

（一）问题描述

105. 从前序与中序遍历序列构造二叉树 - 力扣（LeetCode）105. 从前序与中序遍历序列构造二叉树 - 给定两个整数数组 preorder 和 inorder ，其中 preorder 是二叉树的先序遍历， inorder 是同一棵树的中序遍历，请构造二叉树并返回其根节点。 示例 1:[https://assets.leetcode.com/uploads/2021/02/19/tree.jpg]输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]输出: [3,9,20,null,null,15,7]示例 2:输入: preorder = [-1], inorder = [-1]输出: [-1] 提示: * 1 <= preorder.length <= 3000 * inorder.length == preorder.length * -3000 <= preorder[i], inorder[i] <= 3000 * preorder 和 inorder 均 无重复 元素 * inorder 均出现在 preorder * preorder 保证 为二叉树的前序遍历序列 * inorder 保证 为二叉树的中序遍历序列https://leetcode.cn/problems/construct-binary-tree-from-preorder-and-inorder-traversal/description/?envType=study-plan-v2&envId=top-100-liked给定两个整数数组 preorder 和 inorder ，其中 preorder 是二叉树的先序遍历， inorder 是同一棵树的中序遍历，请构造二叉树并返回其根节点。

示例 1:

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

示例 2:

输入: preorder = [-1], inorder = [-1]
输出: [-1]

提示:

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

（二）解决思路

        解题的关键在于拆解两个数组的结构，两个数组可以这样看：

• 前序遍历preorder=[根节点，[左子树]，[右子树]]
• 中序遍历inorder=[[左子树]，根节点，[右子树]]

        这样就可以采用递归的思想：

• 利用前序遍历数组找到根节点（前序遍历数组的第一个值）
• 找到根节点再中序遍历数组中的位置，进而确定两个数组当中代表左右子树的子序列的起终点
• 创建当前根节点
• 分别处理两个子树

        为了方便找到根节点在中序遍历数组中的位置，可以建立哈希表存储中序遍历数组中元素的值及其索引，避免每次找根节点时都要遍历中序遍历数组。

 

/**
 * 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 {
    private HashMap<Integer,Integer> inHashMap;
    public TreeNode buildTree(int[] preorder, int[] inorder) {     
        //创建哈希表，用于快速定位中序遍历当中根节点的位置
        //这样就不用每次都把中序遍历的数组遍历一遍了
        inHashMap=new HashMap<Integer,Integer>();
        for(int i=0;i<inorder.length;i++){
            inHashMap.put(inorder[i],i);
        }
        int n=preorder.length;
        return build(preorder,inorder,0,n-1,0,n-1);
    }

//in[[左子树]，根节点，[右子树]]
//pre[根节点，[左子树]，[右子树]]
    public TreeNode build(int[] preorder,int[] inorder,int preleft,int preright,int inleft,int inright){
        if(preleft>preright) return null;
        TreeNode root=new TreeNode(preorder[preleft]);
        int idx=inHashMap.get(preorder[preleft]);
        int leftlen=idx-inleft;
//处理左右子树，并与根节点连接
        root.left=build(preorder,inorder,preleft+1,preleft+leftlen,inleft,idx-1);
        root.right=build(preorder,inorder,preleft+leftlen
        +1,preright,idx+1,inright);
        return root;
    }
}

原文地址：https://blog.csdn.net/weixin_45994787/article/details/145291088

免责声明：本站文章内容转载自网络资源，如侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！