N 叉树的层序遍历(力扣刷题)

给定一个 N 叉树，返回其节点值的层序遍历。（即从左到右，逐层遍历）。

树的序列化输入是用层序遍历，每组子节点都由 null 值分隔（参见示例）。

 

提示：

• 树的高度不会超过 1000
• 树的节点总数在 [0, 10^4] 之间

来源：力扣（LeetCode）
链接：https://leetcode.cn/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;
        if(root != NULL)
        {
            que.push(root);
        }     

        vector<vector<int>> result;

        while(!que.empty())
        {
            int size = que.size();
            vector<int> vec;

            for(int i = 0; i < size; i++)
            {
                Node* node = que.front();
                que.pop();
                vec.push_back(node->val);

                for(int i = 0; i < node->children.size(); i++)
                {
                    if(node->children[i])
                    {
                        que.push(node->children[i]);
                    }
                }
            }

            result.push_back(vec);
        }

        return result;
    }
};