Believing Process 剑指Offer7.重建二叉树
作者:互联网
题干:输入某二叉树的前序遍历和中序遍历的结果,请构建该二叉树并返回其根节点。
假设输入的前序遍历和中序遍历的结果中都不含重复的数字。
示例:
Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
Output: [3,9,20,null,null,15,7]
class Solution {
int[] preorder;
HashMap<Integer, Integer> map = new HashMap<>();
// 前序遍历 preorder: 根 -- 左 -- 右 第一个肯定是根节点
// 中序遍历 inorder: 左 -- 根 -- 右
public TreeNode buildTree(int[] preorder, int[] inorder) {
this.preorder = preorder;
for(int i = 0; i < inorder.length; i++){
map.put(inorder[i], i);
}
return rebuild(0, 0, inorder.length - 1);
}
// pre_root_index : 根节点 在 前序遍历中的下标
// in_left_index: 该节点在中序遍历中的左边界
// in_right_index: 该节点在中序遍历中的右边界
public TreeNode rebuild(int pre_root_index, int in_left_index, int in_right_index){
if(in_left_index > in_right_index) return null;
// 根节点在中序遍历中的位置:in_root_index
int in_root_index = map.get(preorder[pre_root_index]);
// 创建一个根节点
TreeNode node = new TreeNode(preorder[pre_root_index]);
// 寻找node的左节点:
// 在前序遍历中的位置就是 根节点的下标 + 1(右边一个单位)
// 在中序遍历中的位置就是: 1. 左边界不变,2. 右边界就是根节点的左边一个单位 in_root_index - 1
node.left = rebuild(pre_root_index + 1, in_left_index, in_root_index - 1);
// 寻找node的右节点:
// 在前序遍历中的位置就是 根节点的下标 + 左子树长度 + 1
// 在中序遍历中的位置就是: 1. 左边界在根节点的右边一个单位 in_root_index + 1, 2. 右边界不变
node.right = rebuild(pre_root_index + in_root_index - in_left_index + 1, in_root_index + 1, in_right_index);
return node;
}
}
理解题目的意思,是要我们根据前序和中序遍历重建二叉树,那么我们能首先想到的是前序遍历的第一个节点就是二叉树的根节点,然后从中序遍历中可以根据根节点,找到左右子树,递归即可完成二叉树的重建。
首先是用一个键值对存储中序遍历中的值以及其对应的下标,方便后续我们可以根据当前节点在前序遍历中的下标从而找到当前节点在中序遍历的中标,这样的一个操作。之后就开可以递归了。
递归三要素,1.终止条件 2.此次递归所需要进行的操作 3.返回值。
根据上述三要素展开递归,首先终止条件就是当前节点的左边界的下标大于当前节点的右边界的下标;此次递归所要进行的操作就是先根据键值对找到当前节点在中序遍历的中下标,然后新建节点,递归去找当前节点的左右子节点;返回值就是当前这个节点。
PS:左子树的长度就是在中序遍历中当前节点的下标 - 左边界下标加1。
标签:index,遍历,Process,中序,前序,Believing,二叉树,root,节点 来源: https://blog.csdn.net/qq_41179409/article/details/120936833