给定一个 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] 之间。
思路
这也是一道普通树类型的题目,不过这题的孩子集合是Node
类型的,但是呢,因为要求最大深度,但是Node
类并不自带层次属性,所以还是得新建一个类,将结点和它本身所在的深度绑定在一起(我这里写的是NodeWithLayer
)。
我这里用了BFS(用DFS也行),每次访问队首元素的时候,查看该结点元素的层次,如果比maxLayer
更大,则更新maxLayer
,当BFS结束后,maxLayer
就是我们要求的最大深度了。
另外,maxLayer
的初始值最好设为0
,这样即便是空树,我们也能返回正确的答案。
代码
public class Solution {
class Node {//实际提交时可以不写类定义
//LeetCode自带
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 NodeWithLayer{
Node node;
int layer;
NodeWithLayer(Node node, int layer){
this.node = node;
this.layer = layer;
}
}
Queue<NodeWithLayer> queue = new LinkedList<NodeWithLayer>();
int maxLayer = 0;
public void bfs(NodeWithLayer root) {
queue.offer(root);
while(queue.isEmpty()==false) {
NodeWithLayer topNode = queue.poll();//出队,并返回队首元素
if(topNode.layer>maxLayer) maxLayer = topNode.layer;//如果当前结点的层次更大,更新
if(topNode.node.children.size()!=0) {//如果有孩子
for(int i=0;i<topNode.node.children.size();i++) {
NodeWithLayer childNode = new NodeWithLayer(topNode.node.children.get(i), topNode.layer+1);
queue.offer(childNode);
}
}
}
}
public int maxDepth(Node root) {
if(root!=null) {
NodeWithLayer rootNode = new NodeWithLayer(root, 1);
bfs(rootNode);
}
return maxLayer;
}
}