题目:
给定一个 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]之间
和二叉树层次遍历类似,使用广度优先搜索算法就可以了。
代码如下:
- /*
- // Definition for a Node.
- class Node {
- public int val;
- public List
children; - public Node() {}
- public Node(int _val) {
- val = _val;
- }
- public Node(int _val, List
_children) { - val = _val;
- children = _children;
- }
- };
- */
-
- class Solution {
- public List
> levelOrder(Node root) {
- List
> res = new ArrayList<>();
-
- Queue
queue = new LinkedList<>(); -
- if(root!=null){
- queue.add(root);
- }
-
- while(!queue.isEmpty()){
- int size = queue.size();
- List
lever = new ArrayList(); - for(int i = 0 ; i < size ; i++){
- Node node = queue.poll();
- lever.add(node.val);
- for(Node child:node.children){
- if(child!=null){
- queue.add(child);
- }
- }
- }
- res.add(lever);
- }
- return res;
- }
- }