链接
https://leetcode-cn.com/problems/n-ary-tree-level-order-traversal/
耗时
解题:13 min
题解:14 min
题意
给定一个 N 叉树,返回其节点值的层序遍历。 (即从左到右,逐层遍历)。
返回其层序遍历:
[
[1],
[3,2,4],
[5,6]
]
思路
正常的 BFS,但是由于需要每层存入一个 vector,所以需要处理一下。
每次遍历一层,为确保每次能遍历一层,用 cnt 记录一下遍历之前队列的长度(即这一层的长度),每访问一个节点就将 cnt-1 直到 cnt == 0,即这一层遍历完成,记录下这一层访问的所有元素,然后下一层同理可得,直到最后一层。
AC代码
/*
// 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) {
if(root == NULL) return {};
queue<Node*> q;
q.push(root);
vector<vector<int>> ans;
while(!q.empty()) {
vector<int> tmp;
int cnt = q.size();
while(cnt) {
Node* r = q.front();
q.pop();
tmp.push_back(r->val);
for(auto c:(r->children)) {
q.push(c);
}
cnt--;
}
ans.push_back(tmp);
}
return ans;
}
};