LeetCode N叉树系列 ～ N叉树的层序遍历

携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第3天，点击查看活动详情

题目

给定一个 N 叉树，返回其节点值的层序遍历。（即从左到右，逐层遍历）。

树的序列化输入是用层序遍历，每组子节点都由 null 值分隔（参见示例）。

示例

输入： root = [1,null,3,2,4,null,5,6]
输出： [[1],[3,2,4],[5,6]]
复制代码

输入：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] 之间

N叉树的前序遍历

树的遍历：

• 前序遍历：首先访问根节点，然后遍历左节点，再遍历右节点；
• 中序遍历：首先遍历左节点，然后访问根节点，再遍历右节点；
• 后序遍历：首先遍历左节点，然后遍历右节点，再访问根节点；
• 层序遍历：按照从左到右的顺序，逐层遍历各个节点。

在二叉树中，有前、中、后序以及层序可以进行遍历，这些遍历的方法，前、后、层序遍历同样可以运用到N叉树中。至于中序遍历，由于并没有一个明确的标准来定义，其中间点因人而异，故暂不考虑N叉树的中序遍历

将二叉树转化成N叉树，我们只需要将其左右节点的遍历修改为对所有子节点的遍历即可，常规的方式有for循环遍历等。

/*
// 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(null == root){
            return res;
        }

        // 队列（先进先出）
        Queue<Node> queue = new LinkedList<>();
        queue.add(root);

        // 遍历每一层的树节点
        while(!queue.isEmpty()){
            // 获取当前层树的节点数
            int size = queue.size();
            
            // 遍历当前层的树节点
            List<Integer> list = new ArrayList<>();
            for(int i = 0; i < size; ++i){
                // 保存节点元素，并将其子节点加入队列中
                Node node = queue.poll();
                list.add(node.val);

                for(Node n : node.children){
                    queue.add(n);
                }
            }
            
            // 保存当前层树节点元素
            res.add(list);
        }

        // 返回结果
        return res;
    }
}
复制代码