【leetcode】429. N叉树的层序遍历（n-ary-tree-level-order-traversal）（BFS）[中等]

链接

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;
    }
};