题目描述
给定一个 N 叉树,找到其最大深度。
最大深度是指从根节点到最远叶子节点的最长路径上的节点总数。
N 叉树输入按层序遍历序列化表示,每组子节点由空值分隔(请参见示例)。
-
示例 1:
输入:root = [1,null,3,2,4,null,5,6]
输出:3 -
示例 2:
输入:root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
输出:5
提示:
树的深度不会超过 1000 。
树的节点数目位于 [0, 104] 之间。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-depth-of-n-ary-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
思路一:DFS,递归。每递归一次,即往更深一层搜索,计数便+1。
思路二:BFS,迭代。每迭代一层树,计数便+1。
代码详解
思路一:DFS
/*
// Definition for a Node.
class Node {
public int val;
public List<Node> children;
public Node() {}
public Node(int _val) {
val = _val;
}
public Node(int _val, List<Node> _children) {
val = _val;
children = _children;
}
};
*/
class Solution {
public int maxDepth(Node root) {
// 也就是最终达成这样一种目的
// return的是每一个子节点的最大深度
// 当递归至根节点root时,便是整个树的最大深度
int max = 0; // 初始化max
if(root == null) {
return 0; // null节点返回0
}
for(Node n : root.children) {
max = Math.max(max, maxDepth(n)); // 对每一个子节点进行递归并取最大值
// 以下代码是错误的
// max = Math.max(max, maxDepth(n)) + 1;
// 然后最后return max
// 不行的原因:在for循环体里每循环一次,max就+1,这显然不正确
}
return max+1; // 此处+1
}
}
思路二:BFS
class Solution {
public int maxDepth(Node root) {
if(root == null) {
return 0;
}
int count = 0;
Queue<Node> queue = new LinkedList<>();
queue.offer(root); // 先把根节点入队
while(!queue.isEmpty()) {
++count; // 此处深度+1
int size = queue.size(); // 因为queue的size()会在下面for循环里改变,因此要记录下来
for(int i = 0; i < size; ++i) {
Node node = queue.poll(); // 当前节点出队后,马上,很快啊
for(Node n : node.children) { // 其孩子节点就入队
queue.offer(n);
}
}
}
return count;
}
}
注意点
- 注意递归时深度何时+1。
- 友情链接:第104题——二叉树的最大深度。