N-ary Tree Level Order Traversal N叉树的层序遍历

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

 

例如，给定一个 3叉树 :

 

 

返回其层序遍历:

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

 

说明:

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

思路：这道题其实就是树的层序遍历，按照之前二叉树的层序遍历进行就行，我们需要一个辅助队列q来保存每层的节点。

参考代码：

/*
// Definition for a Node.
class Node {
public:
    int val = NULL;
    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>> res;
	if (root == nullptr) return res;
	queue<Node*> q;
	q.push(root);
	while (!q.empty()) {
		int n = q.size();
		vector<int> tmp;
		for (int i = 0; i < n; i++) {
			tmp.push_back(q.front()->val);
			int childLength = q.front()->children.size();
			for (int i = 0; i < childLength; i++) {
				q.push(q.front()->children[i]);
			}
			q.pop();
		}
		res.push_back(tmp);
	}
	return res;        
    }
};