Leetcode429. N叉树的层序遍历

题目描述

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

示例：

返回其层序遍历:
[
[1],
[3,2,4],
[5,6]
]

说明：

• 树的深度不会超过 1000。
• 树的节点总数不会超过 5000。

题解

迭代法（java）

思路：用队列处理，在输出列表中添加了当前节点的值之后再把该节点的子节点都添加到队列里。

class Solution {
    public List<List<Integer>> levelOrder(Node root) {
        List<List<Integer>> result = new ArrayList<>();
        if (root == null) return result;
        Queue<Node> queue = new LinkedList<>();
        queue.add(root);
        while(!queue.isEmpty()) {
            List<Integer> list = new ArrayList<>(); 
            int num = queue.size();
            while (num-- > 0) {
                Node cur = queue.poll();
                list.add(cur.val);
                for (Node child : cur.children) {
                    if (child != null) {
                        queue.add(child);
                    }
                }
            }
            result.add(list);
        }
        return result;
    }
}

复杂度分析

• 时间复杂度：$O(n)$ , n是节点数量。
• 空间复杂度：$O(n)$

递归（java）

思路：100%写法。在递归中设置level的概念，然后在不同的递归中，通过level的不同，对输出结果result中不同的list添加元素。

class Solution {
    private List<List<Integer>> result = new ArrayList<>();
![在这里插入图片描述](https://img-blog.csdnimg.cn/20200326234920399.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0NoZW5fMjAxOGs=,size_16,color_FFFFFF,t_70)
    public List<List<Integer>> levelOrder(Node root) {
        if (root != null) traverseNode(root, 0);
        return result;
    }

    private void traverseNode(Node node, int level) {
        if (result.size() <= level) {
            result.add(new ArrayList<>());
        }
        result.get(level).add(node.val);
        for (Node child : node.children) {
            traverseNode(child, level + 1);
        }
    }
}

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$