重建二叉树(算法题)
Published in:2022-07-23 | category: 算法

题目:

        输入某二叉树的前序遍历和中序遍历的结果,请构建该二叉树并返回其根节点。

        假设输入的前序遍历和中序遍历的结果中都不含重复的数字。

示例:

        Input: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7]
        Output: [3,9,20,null,null,15,7]

代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
/**
* 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> in=new HashMap<>();
public TreeNode buildTree(int[] preorder, int[] inorder) {
this.preorder=preorder;
for(int i=0;i<inorder.length;i++){
in.put(inorder[i],i); //方便获取根节点在中序遍历的坐标
}
return hui(0,0,inorder.length-1);
}
TreeNode hui(int root,int left,int right){ //递归方法
if(left>right) //递归结束的条件
return null;
TreeNode node=new TreeNode(preorder[root]); //创建节点
int index=in.get(preorder[root]); //获取根节点坐标
node.left=hui(root+1,left,index-1); //获取左子树
node.right=hui(root+index-left+1,index+1,right);
//root+index-left+1为当前节点+左子树个数+1
return node;
}
}

** 总结:**

        前序遍历时先是根节点,然后是左子树,右子树。

        中序遍历先是左子树,然后是根节点,最后是右子树。 

        前序遍历第一个数是二叉树最上面的根节点,遍历inorder,找到根节点的位置。根节点左边是左子树的中序遍历结果,右边是右子树的遍历结果。先找左子树,在左子树的遍历结果里最上面的一个根节点就是前序遍历的第二个数,以此类推,采用递归的方式解决问题。

        这是第一次接触到有关树的算法题,开始的时候题目都没有怎么看懂,后来看题解还是有好多地方感觉有点疑问。后来花费了好多时间才把这道题理解个差不多,记录一下,希望后来再遇到这种类型的题目时我可以自己写出来。

题目链接:https://leetcode.cn/problems/zhong-jian-er-cha-shu-lcof

Prev:
对称的二叉树(算法题)
Next:
Mybatis-plus常用方法