题目要求
给定一个N叉树,返回树的最大深度。以一个三叉树为例。 输出应为3。
解题思路
这个问题和其他的N叉树问题类似,试想一下,求解最大深度的问题,我们在二叉树的时候也有说,求最大深度的时候,他和求树的层数是等价的,树的层数即为树的最大深度(如上图),那么涉及到层的问题,我们想到可以用队列进行迭代,N叉树和二叉树不同的还是孩子的个数,所以用for(auto child:root->children) 保证每个子节点都遍历到。
主要代码 c++
/*
// Definition for a Node.
class Node {
public:
int val;
vector<Node*> children;
Node() {}
Node(int _val, vector<Node*> _children) {
val = _val;
children = _children;
}
};
*/
class Solution {
public:
int maxDepth(Node* root) {
if(root==NULL) return 0;
int level=0; // 记录层数
queue<Node*>q;
q.push(root);
while(!q.empty()) // 迭代条件
{
level +=1;
int size = q.size();
for(int i=0;i<size;++i)
{
Node *tmp=q.front(); q.pop();
for(auto child:tmp->children) //遍历所有孩子
q.push(child);
}
}
return level;
}
};
划重点:
有关N叉树的重点是如何遍历所有的孩子节点,掌握这个,其他的问题可以引用二叉树的思想去解答
相似题目:
解答:leetcode 429 N-ary Tree Level Order Traversal(N叉树的层序遍历)
解答:leetcode 589 N-ary Tree Preorder Traversal(N叉树的前序遍历)
解答:leetcode 590 N-ary Tree Postorder Traversal(N叉树的后序遍历)