题目描述
给定一个 N 叉树,返回其节点值的层序遍历。(即从左到右,逐层遍历)。
树的序列化输入是用层序遍历,每组子节点都由 null 值分隔(参见示例)。
示例 1:
输入:root = [1,null,3,2,4,null,5,6]
输出:[[1],[3,2,4],[5,6]]
示例 2:
输入:root = [1,null,2,3,4,5,null,null,6,7,null,8,null,9,10,null,null,11,null,12,null,13,null,null,14]
输出:[[1],[2,3,4,5],[6,7,8,9,10],[11,12,13],[14]]
提示:
树的高度不会超过 1000
树的节点总数在 [0, 10^4] 之间
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/n-ary-tree-level-order-traversal
方法一:
广度优先搜索,队列, 后面发现每一层的结点数可以用qu.size()得到,由于太懒就不改了
/*
// 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) {
int n=0,f=1,j=0;
queue<Node*> qu;
vector<vector<int>> ans;
vector<int> res;
if(root){
qu.emplace(root);
}
while(!qu.empty()){
auto t=qu.front();
qu.pop();
res.push_back(t->val);
for(int k=0;k<t->children.size();k++){
qu.emplace(t->children[k]);
j++;
}
n++;
if(n==f){
f=j;
n=0;
j=0;
ans.push_back(res);
res.erase(res.begin(),res.end());
}
}
return ans;
}
};