题目[题目及部分解法来源于力扣]
给定两个整数数组 preorder 和 inorder ,其中 preorder 是二叉树的先序遍历, inorder 是同一棵树的中序遍历,请构造二叉树并返回其根节点。
solution one :
首先要了解什么是先序遍历,中序遍历!- 先序遍历:先遍历根节点;随后递归地遍历左子树;最后递归地遍历右子树。
- 后序遍历:先递归地遍历左子树;随后遍历根节点;最后递归地遍历右子树。
根据遍历顺序,先序遍历一定遍历根节点,我们就可以知道前序遍历第一个节点一定是根节点。同时,再对应到中序遍历序列中,中序遍历序列左侧即为左子树节点,右侧即为右子树节点。由于同一颗子树的前序遍历和中序遍历的长度显然是相同的,这样我们就可以确定左、右子树节点数目,因此我们就可以对应到前序遍历的结果中,对上述形式中的所有左右括号进行定位。同时,也得到了左、右侧子树遍历的结果,就可以递归地构造出左子树和右子树。
public class buildTrees1 {
public int preIndex;
public int findInOrderIndex(int target,int[] inorder,int begin,int end){ //用来查找根节点在中序序列中的位置
for(int i=begin;i<=end;i++){
if(inorder[i]==target)
return i;
}
return -1;
}
public TreeNode createTree(int[] preorder,int[] inorder,int begin,int end){
// 递归结束条件
if(begin>end) return null;
//找到中序遍历中的根节点的下标
int inIndex=findInOrderIndex(preorder[preIndex],inorder,begin,end);
//如果,没有找到,就返回null
if(inIndex==-1) return null;
//如果找到了,就创建根节点
TreeNode root=new TreeNode(preorder[preIndex]);
//再分别去递归遍历左子树和右子树
preIndex++;
root.left=createTree(preorder,inorder,begin,inIndex-1); //这里定位的结果,左子树一定是从begin~index -1(根节点在中序序列中的位置的前一个节点)
root.right=createTree(preorder,inorder,inIndex+1,end); //这里定位的结果,右子树一定是从index +1~end(根节点在中序序列中的位置的后一个节点)
return root;
}
public TreeNode buildTree(int[] preorder, int[] inorder) {
if(preorder==null||inorder==null) return null; //数为空,返回null
return createTree(preorder,inorder,0,preorder.length-1);
}
}
solution two:
在solution one中,我们是使用for循环来遍历中序序列,找到中序序列中的根节点,进而确定左右子树范围,然而这里的时间复杂度为O(n)。在这里我们使用**哈希表**进行改进,帮助我们快速定位根节点。- **对于哈希映射中的每个键值对,键表示一个元素(节点的值),值表示其在中序遍历中的出现位置。**在构造二叉树的过程之前,我们可以对中序遍历的列表进行一遍扫描,就可以构造出这个哈希映射。在此后构造二叉树的过程中,我们就只需要 O(1) 的时间对根节点进行定位了。
public class buildTrees {
private Map<Integer, Integer> indexMap; //声明hashMap
public TreeNode myBuildTree(int[] preorder, int[] inorder, int preorder_left, int preorder_right, int inorder_left, int inorder_right) {
if (preorder_left > preorder_right) {
return null;
}
// 前序遍历中的第一个节点就是根节点
int preorder_root = preorder_left;
// 在中序遍历中定位根节点
int inorder_root = indexMap.get(preorder[preorder_root]);
// 先把根节点建立出来
TreeNode root = new TreeNode(preorder[preorder_root]);
// 得到左子树中的节点数目
int size_left_subtree = inorder_root - inorder_left;
// 递归地构造左子树,并连接到根节点
// 先序遍历中「从 左边界+1 开始的 size_left_subtree」个元素就对应了中序遍历中「从 左边界 开始到 根节点定位-1」的元素
root.left = myBuildTree(preorder, inorder, preorder_left + 1, preorder_left + size_left_subtree, inorder_left, inorder_root - 1);
// 递归地构造右子树,并连接到根节点
// 先序遍历中「从 左边界+1+左子树节点数目 开始到 右边界」的元素就对应了中序遍历中「从 根节点定位+1 到 右边界」的元素
root.right = myBuildTree(preorder, inorder, preorder_left + size_left_subtree + 1, preorder_right, inorder_root + 1, inorder_right);
return root;
}
public TreeNode buildTree(int[] preorder, int[] inorder) {
int n = preorder.length;
// 构造哈希映射,帮助我们快速定位根节点
indexMap = new HashMap<Integer, Integer>();
for (int i = 0; i < n; i++) {
indexMap.put(inorder[i], i);
}
return myBuildTree(preorder, inorder, 0, n - 1, 0, n - 1);
}
}
***!!!HashMap知识点复盘,熟悉的小伙伴阔以直接redirect了!!!*** Java HashMap
- HashMap 是一个散列表,它存储的内容是键值对(key-value)映射。
- HashMap 实现了 Map 接口,根据键的 HashCode 值存储数据,具有很快的访问速度,最多允许一条记录的键为 null,不支持线程同步。
- HashMap 是无序的,即不会记录插入的顺序。
- HashMap 继承于AbstractMap,实现了 Map、Cloneable、java.io.Serializable 接口。
- HashMap 的 key 与 value 类型可以相同也可以不同,可以是字符串(String)类型的 key 和 value,也可以是整型(Integer)的 key 和字符串(String)类型的 value。
- 注意,HashMap 中的元素实际上是对象,一些常见的基本类型可以使用它的包装类,如Boolean、Integer、Byte、Short、Long、Float、Double、Character。
- HashMap类常用的方法:
- 添加键值对(key, value)可以使用 put() 方法
- get(key) 方法来获取 key 对应的 value
- remove(key) 方法来删除 key 对应的键值对(key-value)
- 删除所有键值对(key-value)可以使用 clear 方法
- HashMap 中的元素数量可以使用 size() 方法:
- 可以使用 for-each 来迭代 HashMap 中的元素。如果你只想获取 key,可以使用 keySet() 方法,然后可以通过 get(key) 获取对应的 value,如果你只想获取 value,可以使用 values() 方法。