LeetCode_19_NAryTreeLevelOrderTraversal

1. question: 二叉树的层序遍历（中等）

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/n-ary-tree-level-order-traversal

示例 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] 之间

2. answer

这道题和二叉树的层序遍历一样，只是在入栈的时候，将其孩子List遍历一遍入栈而已。代码如下所示：

public 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;
    }
}

public class Solution_0017 {

    public static List<List<Integer>> levelOrder(Node root) {

        List<List<Integer>> result = new ArrayList<>();

        if(root == null) {
            return result;
        }

        Queue<Node> queue = new LinkedList<>();
        List<Integer> subList;

        queue.offer(root);

        int length;

        while(!queue.isEmpty()) {
            length = queue.size();
            subList = new ArrayList<>();

            while(length > 0) {
                root = queue.poll();
                subList.add(root.val);

                // 将孩子List入栈
                if(root.children != null) {
                    queue.addAll(root.children);
                }

                length --;
            }

            result.add(subList);
        }

        return result;
    }

    public static void main(String[] args) {

        Node n1 = new Node(5);
        Node n2 = new Node(6);

        List<Node> l1 = new ArrayList<>();
        l1.add(n1);
        l1.add(n2);

        Node n3 = new Node(3, l1);
        Node n4 = new Node(2);
        Node n5 = new Node(4);

        List<Node> l2 = new ArrayList<>();
        l2.add(n3);
        l2.add(n4);
        l2.add(n5);

        Node root = new Node(1, l2);

        List<List<Integer>> result = levelOrder(root);

        for(List<Integer> l: result) {

            for(Integer i: l) {
                System.out.print(i + " ");
            }
            System.out.println();
        }

    }
}

3. 备注

参考力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台 (leetcode-cn.com)，代码随想录 (programmercarl.com)。