一、需求
-
输入一棵二叉树的根节点,求该树的深度。
-
从根节点到叶节点依次经过的节点(含根、叶节点)形成树的一条路径,最长路径的长度为树的深度。
二、后序遍历法(DFS)
2.1 思路分析
- 递归左子树深度、递归右子树深度,找到其中的最大值,然后加上当前节点对应的深度1;
2.2 代码实现
class Solution {
public int maxDepth(TreeNode root) {
if(root == null) return 0;
return Math.max(maxDepth(root.left),maxDepth(root.right)) + 1;
}
}
2.3 复杂度分析
- 时间复杂度为O(N),N为树的节点数量,计算树的深度需要遍历所有的节点;
- 空间复杂度为O(N),最差情况下,树退化为链表,递归深度可达N;
三、层序遍历法(BFS)
3.1 思路分析
- 二叉树的层序遍历往往借助队列实现,使用LinkedList来模拟队列;
- 队列用来存储每一层的节点,当队列不为空时,遍历队列,将队列中各节点的左右子节点加入到一个新队列中,然后将队列换成新队列,直到队列为空时,退出循环;
3.2 代码实现1
class Solution {
public int maxDepth(TreeNode root) {
if(root == null) return 0;
List<TreeNode> queue = new LinkedList<>();
int res = 0;
queue.add(root);
while(queue.size() != 0) {
res++;
List<TreeNode> tmp = new LinkedList<>();
for(TreeNode node : queue) {
if(node.left != null) tmp.add(node.left);
if(node.right != null) tmp.add(node.right);
}
queue = tmp;
}
return res;
}
}
3.3 复杂度分析
- 时间复杂度为O(N),N为树的节点数量,计算树的深度需要遍历所有的节点;
- 空间复杂度为O(N),最差情况下,当树平衡时,队列最多存储N/2个元素,即最后一行的最大个数;
四、学习地址
作者:Krahets