给定一个 N 叉树,返回其节点值的层序遍历。 (即从左到右,逐层遍历)。
例如,给定一个 3叉树 :
返回其层序遍历:
[
[1],
[3,2,4],
[5,6]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/n-ary-tree-level-order-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:层序遍历,不是向队列中插入该节点的左右子树,是插入该节点的孩子。
/*
// Definition for a Node.
class Node {
public:
int val;
vector<Node*> children;
Node() {}
Node(int _val) {
val = _val;
}
Node(int _val, vector<Node*> _children) {
val = _val;
children = _children;
}
};
*/
class Solution {
public:
vector<vector<int>> levelOrder(Node* root) {
queue<Node*>que;
vector<vector<int>>re;
Node* node;
if(root!=nullptr) que.push(root);
while(!que.empty())
{
int size=que.size();
vector<int>res;
for(int i=0;i<size;i++)
{
node=que.front();
que.pop();
res.push_back(node->val);
for(int j=0;j<node->children.size();j++)
{
que.push(node->children[j]);
}
}
re.push_back(res);
}
return re;
}
};