经典面试题: 力扣 重构二叉树——java 分治递归算法。
剑指 Offer 07. 重建二叉树 输入某二叉树的前序遍历和中序遍历的结果,请构建该二叉树并返回其根节点。
假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
示例 1:
Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]
示例 2:
Input: preorder = [-1], inorder = [-1]
Output: [-1]
(前 中 后 根遍历, 即 遍历根的优先级)
- 前序遍历的性质: 根节点 左子树 右子树
- 中序遍历的性质: 左子树 根节点 右子树
- 后序遍历的性质: 左子树 右子树 很节点
由树遍历的性质,可以得到 前序遍历的第一个元素是树的根节点。
中序遍历 将 二叉树的遍历结果分为两部分,以根节点为中心,左半部分是左子树
右半部分是右子树。 而根据左子树,右子树的长度,进一步细分前序遍历。
进而得到根节点,左子树根节点,右子树根节点的位置(索引)。
进而再通过上述的算法,递归分治
把左子树和右子树 根节点,子树节点找出来。构成完成二叉树序列。
分治算法解析:
递归参数: 根节点在前序遍历的索引root,
子树在中序遍历的左边界left,子树在中序遍历的右边界right。
终止条件: 当 left > right 时,说明越界。
建立根节点 node : 节点值为 preorder[root] ;
划分左右子树: 查找根节点在中序遍历 inorder 中的索引 i ;
为了提升效率,使用哈希表 dic 存储中序遍历的值与索引的映射,查找操作的时间复杂度为 O(1)O(1) ;
构建左右子树: 开启左右子树递归;
返回值: 回溯返回 node ,作为上一层递归中根节点的左 / 右子节点;
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
int [] preorder ;
HashMap <Integer,Integer> map = new HashMap<>();
public TreeNode buildTree(int[] preorder, int[] inorder) {
this.preorder = preorder;
for(int i =0; i<inorder.length;i++){
// 把中序遍历的结果到hashmap。
map.put(inorder[i],i);
}
//递归
return recur(0,0,inorder.length-1);
}
// 三个参数分明是根节点,左边界,右边界
TreeNode recur(int root, int left, int right){
if(left>right)return null;
//初始根节点
TreeNode node = new TreeNode(preorder[root]);
//获取根节点的索引,划分左右子树
int indexRoot = map.get(preorder[root]);
// 左右递归。
node.left = recur(root+1,left,indexRoot -1);
// root +i -left +1 为右子树的根节点。
node.right = recur(root+ indexRoot - left +1,indexRoot +1,right);
// 回溯返回根节点
return node ;
}
}