Offer系列面试题0：重建二叉树-电子发烧友网

今天分享的题目来源于 LeetCode 上的剑指 Offer 系列面试题07. 重建二叉树，近半年在微软面试环节出现过 2 次，属于中高难度的算法题！

一、题目描述

输入某二叉树的前序遍历和中序遍历的结果，请重建该二叉树。假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

例如，给出

前序遍历preorder=[3,9,20,15,7] 中序遍历inorder=[9,3,15,20,7]

返回如下的二叉树：

3 / 920 / 157

限制：

0 <= 节点个数 <= 5000

二、题目解析

首先，我们先来复习一下前序遍历、中序遍历。（在下方的视频中分布讲解）

前序遍历

二叉树的前序遍历顺序是：根节点、左子树、右子树，每个子树的遍历顺序同样满足前序遍历顺序。

中序遍历

二叉树的中序遍历顺序是：左子树、根节点、右子树，每个子树的遍历顺序同样满足中序遍历顺序。

复习过后，我们可以得出以下结论：

在二叉树的前序遍历序列中，第一个数字总是树的根结点的值；

在二叉树的中序遍历序列中，根结点的值在序列的中间，左子树的结点的值位于根结点的值的左边，而右子树的结点的值位于根结点的值的右边

以本题的序列为例，前序遍历序列的第一个数字 3 就是根结点的值，在中序遍历序列，找到根结点值的位置。根据中序遍历特点，在根结点的值 3前面的数字都是左子树结点的值，在根结点的值 3后面的数字都是右子树结点的值。

二叉树很重要的一个性质是递归，在找到了左子树、右子树的前序遍历序列和中序遍历序列后，我们可以按照同样的方法去确定子左子树和子右子树的构建。

具体的代码编写思路如下（来源于 Krahets's Blog）：

递推参数：前序遍历中根节点的索引pre_root_idx、中序遍历左边界in_left_idx、中序遍历右边界in_right_idx。

终止条件：当in_left_idx > in_right_idx，子树中序遍历为空，说明已经越过叶子节点，此时返回 null 。

递推工作：

建立根节点 root ：值为前序遍历中索引为pre_root_idx的节点值。

搜索根节点 root 在中序遍历的索引 i ：为了提升搜索效率，本题解使用哈希表map预存储中序遍历的值与索引的映射关系，每次搜索的时间复杂度为 O(1)。

构建根节点root的左子树和右子树：通过调用 recursive()方法开启下一层递归。

左子树：根节点索引为 pre_root_idx + 1 ，中序遍历的左右边界分别为 in_left_idx 和 i - 1。

右子树：根节点索引为 i - in_left_idx + pre_root_idx + 1（即：根节点索引 + 左子树长度 + 1），中序遍历的左右边界分别为 i + 1 和 in_right_idx。

返回值：返回root，含义是当前递归层级建立的根节点root为上一递归层级的根节点的左或右子节点。

三、动画描述

四、图片描述

五、参考代码

classSolution{ //在中序序列中查找与前序序列首结点相同元素的时候，如果使用while循环去一个个找效率很慢 //这里我们借助数据结构HashMap来辅助查找，在开始递归之前把所有的中序序列的元素和它们所在的下标存到一个map中，这样查找的时间复杂度是O(logn) HashMapmap=newHashMap<>(); //保留的前序遍历 int[]preorder; publicTreeNodebuildTree(int[]preorder,int[]inorder){ this.preorder=preorder; //在开始递归之前把所有的中序序列的元素和它们所在的下标存到一个map中 for(inti=0;i< preorder.length; i++) { map.put(inorder[i], i); } //二叉树的重要性质是递归 return recursive(0,0,inorder.length-1); } /** 根据前序遍历序列和中序遍历序列重新组建二叉树 * @param pre_root_idx 前序遍历的索引 * @param in_left_idx 中序遍历左边界的索引 * @param in_right_idx 中序遍历右边界的索引 */ public TreeNode recursive(int pre_root_idx, int in_left_idx, int in_right_idx) { //子树中序遍历为空，说明已经越过叶子节点，此时返回 nul if (in_left_idx >in_right_idx){ returnnull; } //root_idx是在前序里面的 TreeNoderoot=newTreeNode(preorder[pre_root_idx]); //通过map，根据前序的根节点的值，在中序中获取当前根的索引 intidx=map.get(preorder[pre_root_idx]); //左子树的根节点就是左子树的(前序遍历）第一个，就是+1,左边边界就是left，右边边界是中间区分的idx-1 root.left=recursive(pre_root_idx+1,in_left_idx,idx-1); //右子树的根，就是右子树（前序遍历）的第一个,就是当前根节点加上左子树的数量 root.right=recursive(pre_root_idx+(idx-1-in_left_idx+1)+1,idx+1,in_right_idx); returnroot; } }

这段代码的一个难点就是root.left与root.right，我这里抽离出来详细解释一下。

1、root.left