429. N 叉树的层序遍历
给定一个 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 , 1 0 4 ] [0, 10^4] [0,104]之间
解法一(BFS+队列)
思路分析:
- 虽然该树不是二叉树,但是依然可以使用广度优先搜索进行层序遍历
实现代码如下:
class Solution {
public List<List<Integer>> levelOrder(Node root) {
List<List<Integer>> ans = new ArrayList<>();
if (root == null)
return ans; // 考虑边界条件
Queue<Node> queue = new ArrayDeque<>();
// 层序遍历
queue.offer(root);
while (!queue.isEmpty()) {
int size = queue.size();
List<Integer> level = new ArrayList<>(); // 记录该层结点值
for (int i = 0; i < size; ++ i) {
Node node = queue.poll();
level.add(node.val); // 保存记录该层结点
List<Node> children = node.children; // 记录该层孩子结点
// if (children != null) {
// children.forEach(queue::offer); // 将该孩子的结点进队 用于下一层遍历
// }
for (Node child : children) { // 增强for 比 lambda表达式开销更低
queue.offer(child);
}
}
ans.add(level);
}
return ans;
}
}
提交结果如下:
解答成功:
执行耗时:3 ms,击败了88.49% 的Java用户
内存消耗:43.6 MB,击败了9.00% 的Java用户
复杂度分析:
- 时间复杂度: O ( n ) O(n) O(n)
- 空间复杂度: O ( n ) O(n) O(n)