力扣题目
解题思路
java代码
力扣题目:
定两个整数数组 preorder
和 inorder
,其中 preorder
是二叉树的先序遍历, inorder
是同一棵树的中序遍历,请构造二叉树并返回其根节点。
示例 1:
输入: preorder = [3,9,20,15,7], inorder = [9,3,15,20,7] 输出: [3,9,20,null,null,15,7]
示例 2:
输入: preorder = [-1], inorder = [-1] 输出: [-1]
解题思路:
算法原理:
利用前序遍历和中序遍历的特性来重建二叉树。前序遍历的第一个节点是根节点,在中序遍历中找到根节点的位置,就可以确定左子树和右子树的节点范围,然后递归地构建左右子树。
思路:
- 首先判断前序遍历数组是否为空,如果为空则返回
null
。 - 从前序遍历数组中取出第一个元素作为根节点的值创建根节点。
- 在中序遍历数组中找到根节点的值的位置,该位置将中序遍历数组分为左子树节点和右子树节点。
- 根据中序遍历中根节点的位置,对前序遍历数组进行分割,得到左子树和右子树的前序遍历子数组。同时,对中序遍历数组也进行相应分割,得到左子树和右子树的中序遍历子数组。
- 递归地使用左子树的前序和中序子数组构建左子树,使用右子树的前序和中序子数组构建右子树,并将构建好的左右子树连接到根节点上。
代码分析:
- 在
buildTree
方法中,首先处理前序遍历数组为空的情况。 - 创建根节点,并通过循环在中序遍历数组中找到根节点的位置。
- 然后通过数组切片的方式获取左右子树的前序和中序遍历子数组,并递归构建左右子树。
时间复杂度:O(n),其中是节点的数量。因为每个节点都会被访问一次。
空间复杂度:O(n),主要是递归调用栈的空间开销。在最坏情况下,树可能退化为链表,递归深度达到。
java代码:
package mouth8;
import java.util.Arrays;
public class Leetcode105 {
public static void main(String[] args) {
int[] preorder = {3, 9, 20, 15, 7};
int[] inorder = {9, 3, 15, 20, 7};
TreeNode treeNode = buildTree(preorder, inorder);
System.out.println(treeNode);
}
public static TreeNode buildTree(int[] preorder, int[] inorder) {
if (preorder.length == 0) {
return null;
}
TreeNode root = new TreeNode(preorder[0]);
int index = 0;
for (int i = 0; i < inorder.length; i++) {
if (inorder[i] == preorder[0]) {
index = i;
break;
}
}
root.left = buildTree(Arrays.copyOfRange(preorder, 1, index + 1), Arrays.copyOfRange(inorder, 0, index));
root.right = buildTree(Arrays.copyOfRange(preorder, index + 1, preorder.length), Arrays.copyOfRange(inorder, index + 1, inorder.length));
return root;
}
}
更多详细内容同步到公众号,感谢大家的支持!
每天都会给刷算法的小伙伴推送明日一题,并且没有任何收费项