429N叉树的层序遍历

N叉树的层序遍历

题目描述：

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

例如，给定一个 3叉树 :

返回其层序遍历:

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

说明:

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

解题思路：本题和之前的二叉树层次遍历思路一样，只不过就是孩子结点的数目不一样，还有要注意的就是输出的这种形式，使用当前队列的长度这一点很巧妙（正好对应这当前层次的节点数目），最后将vector插入到二维的vector即可。

class Solution {
public:
    vector<vector<int>> levelOrder(Node* root) {
        vector<int>layers;
        vector<vector<int>>result;
        if(!root)return result;
        queue<Node*>node;
        Node* p;
        node.push(root);
        int cur_size;
        while(!node.empty()){
            cur_size=node.size();
            for(int i=0;i<cur_size;i++){
                p=node.front();
                layers.push_back(p->val);
                node.pop();
                for(int j=0;j<p->children.size();j++){
                    if(p->children[j])node.push(p->children[j]);
                }
            }
            result.push_back(layers);
            layers.clear();
        }
        return result;
    }
};