给定一个 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]之间
解题思路:队列 + 广度搜索
这道题其实就是 二叉树的层序遍历 的变形,只不过将其左右孩子改成了一个数组来存放孩子节点罢了,我们只需要遍历一下数组即可,其它思路都是一样的,就是使用队列的先进先出特点,每次处理一层,然后根据队列的元素个数控制每层遍历d
/*
// 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) {
if(root == nullptr)
return {};
vector<vector<int>> vv; // 结果集
queue<Node*> qe; // 队列,存放的是树的节点
qe.push(root);
while(!qe.empty())
{
int n = qe.size();
vector<int> v;
for(int i = 0; i < n; ++i)
{
Node* front = qe.front();
qe.pop();
v.push_back(front->val);
for(int j = 0; j < front->children.size(); ++j) // 将当前节点的所有子节点入队列
qe.push(front->children[j]);
}
vv.push_back(v); // 别忘了尾插到结果集中
}
return vv;
}
};