429. N叉树的层序遍历

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

例如，给定一个 3叉树 :

返回其层序遍历:

[
[1],
[3,2,4],
[5,6]
]
 

说明:

树的深度不会超过 1000。
树的节点总数不会超过 5000。

solution:

/*
// 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:
    vector<vector<int>> levelOrder(Node* root) {
        vector<vector<int> >vec;
        if(root)
        {
          queue<Node*>q;
          q.push(root);
          while(!q.empty())
          {
            int size = q.size();
            vector<int> vecChild;
            while(size--)
            {
              Node* node = q.front();
              q.pop();
              vecChild.push_back(node->val);
              for(auto child : node->children)
                q.push(child);
            }
            vec.push_back(vecChild);
          } 
        }
      return vec;
    }
};