给定二叉搜索树的插入顺序求深度
难度:
标签:
题目描述
代码结果
运行时间: 2055 ms, 内存: 39.4 MB
/*
* 题目思路:
* 使用Java Stream的方式插入节点,并计算二叉搜索树的最大深度。
* 我们使用递归方式计算最大深度。
*/
import java.util.Arrays;
public class Solution {
// 定义树节点类
static class TreeNode {
int val;
TreeNode left, right;
TreeNode(int x) { val = x; }
}
// 插入节点到二叉搜索树
public TreeNode insert(TreeNode root, int val) {
if (root == null) {
return new TreeNode(val);
}
if (val < root.val) {
root.left = insert(root.left, val);
} else if (val > root.val) {
root.right = insert(root.right, val);
}
return root;
}
// 计算树的最大深度
public int maxDepth(TreeNode root) {
if (root == null) return 0;
return 1 + Math.max(maxDepth(root.left), maxDepth(root.right));
}
// 主函数
public int getDepth(int[] nums) {
// 使用Stream API构建BST并计算最大深度
TreeNode root = Arrays.stream(nums).reduce(null, this::insert, (a, b) -> a);
return maxDepth(root);
}
}
解释
方法:
该题解使用了一个平衡二叉树(通过SortedList实现)和一个哈希表(defaultdict)来跟踪插入的每个元素的深度。对于输入的每一个元素x,我们首先查找它应该插入的位置。在SortedList中,我们找到紧邻x左侧的元素k,并将x的深度设置为k的深度加1。每次插入后,我们将x添加到SortedList中以保持元素的有序性,并更新最大深度。通过这种方式,我们可以有效地计算出二叉搜索树的最大深度。
时间复杂度:
O(n log n)
空间复杂度:
O(n)
代码细节讲解
🦆
为什么选择使用SortedList而不是其他数据结构如AVL树或红黑树来维护元素的有序性?
▷🦆
在寻找x左侧的元素k时,如果x小于SortedList中的所有元素,这种情况下如何处理?
▷🦆
代码中`book[x + 1] = book[k + 1] = book[k + 1] + 1`似乎有语法错误或逻辑错误,请问正确的更新方式应该是什么?
▷