题目描述
输入一棵二叉树的根节点,求该树的深度。从根节点到叶节点依次经过的节点(含根、叶节点)形成树的一条路径,最长路径的长度为树的深度。
例如:
给定二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回它的最大深度 3 。提示:节点总数 <= 10000
参考解题思路:深度遍历(DFS),递归比较二叉树的左节点和右节点的深度那边更大,再加1返回即可
算法解析:
- 终止条件: 当 root 为空,说明已越过叶节点,因此返回 深度 0 。
- 递推工作: 本质上是对树做后序遍历。
- 计算节点 root 的 左子树的深度 ,即调用 maxDepth(root.left);
- 计算节点 root 的 右子树的深度 ,即调用 maxDepth(root.right);
- 返回值: 返回此树的深度 ,即 max(maxDepth(root.left), maxDepth(root.right)) + 1。
public int maxDepth(TreeNode root) {
if (root == null) {
return 0;
}
return Math.max(maxDepth(root.left), maxDepth(root.right)) + 1;
}
复杂度分析:
时间复杂度 O(N) : N为树的节点数量,计算树的深度需要遍历所有节点。
空间复杂度 O(N): 最差情况下(当树退化为链表时),递归深度可达到 N。
参考解题思路二:层序遍历(BFS),利用队列存储每层节点,再将节点的左右子节点按序追加进队列,每遍历一层,层序记录加一,最后返回所记录层序值即可
算法解析:
- 特例处理: 当 root 为空,直接返回 深度 0 。
- 初始化: 队列 queue (加入根节点 root ),计数器 res = 0。
- 循环遍历: 当 queue 为空时跳出。
- 遍历队列: 遍历 queue 中的各节点 node ,并将其左子节点和右子节点加入queue;
- 统计层数: 执行 res += 1 ,代表层数加 1;
- 返回值: 返回 res 即可。
public int maxDepth(TreeNode root) {
Queue<TreeNode> queue = new LinkedList<>(); // 定义队列
// 如果root不为空则将根节点加入到队列中
if (root != null) {
queue.add(root);
}
int res = 0; // 统计层数
while (!queue.isEmpty()) {
res++; // 层数统计加一
int n = queue.size(); // 该层节点数量,防止遍历节点时动态变化
for (int i = 0; i < n; i++) {
TreeNode node = queue.poll(); // 取出队首节点
// 该节点的左节点不为空则加入队列中
if (node.left != null) {
queue.add(node.left);
}
// 该节点的右节点不为空则加入到队列中
if (node.right != null) {
queue.add(node.right);
}
}
}
return res;
}
复杂度分析:
时间复杂度 O(N): N 为树的节点数量,计算树的深度需要遍历所有节点。
空间复杂度 O(N): 最差情况下(当树平衡时),队列 queue 同时存储 N/2 个节点。
作者:jyd
链接:https://leetcode-cn.com/problems/er-cha-shu-de-shen-du-lcof/solution/mian-shi-ti-55-i-er-cha-shu-de-shen-du-xian-xu-bia/