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, 104] 之间
思路:广度优先遍历。定义一个队列,还有一个cnt计数器,cnt记录每一层的节点数,其实就是进入该层遍历时队列的大小,然后根据cnt循环弹出,放到一个临时结果集,跳出循环后,把这一层结果加入到结果集
细节:
需要特殊判断树为空的情况。
/*
// Definition for a Node.
class Node {
public int val;
public List<Node> children;
public Node() {}
public Node(int _val) {
val = _val;
}
public Node(int _val, List<Node> _children) {
val = _val;
children = _children;
}
};
*/
class Solution {
public List<List<Integer>> levelOrder(Node root) {
List<List<Integer>> res = new ArrayList<>();
if(root==null){
return res;
}
Queue<Node> queue = new LinkedList<>();
queue.add(root);
int cnt;
while(!queue.isEmpty()){
cnt=queue.size();
List<Integer> list = new ArrayList<>();
while(cnt>0){
Node node = queue.poll();
list.add(node.val);
cnt--;
if(node.children.size()>0){
for(Node n:node.children){
queue.add(n);
}
}
}
res.add(list);
}
return res;
}
}