429. N 叉树的层序遍历

188 阅读2分钟

这是我参与11月更文挑战的第3天,活动详情查看:2021最后一次更文挑战

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, 10^4] 之间

解题思路

我们使用队列实现层序遍历,因为如果使用 ArrayList 的话,我们删除元素需要 O(n) 的时间复杂度。而队列删除元素只需要 O(1)的时间。

并且在while 循环体开始时记录队列的当前大小 size。然后用另一个循环来处理 size 数量的节点,这样我们就可以保证我们每次使队列出队的都是位于同一层的节点,并且后面入队的节点也全是位于当前层的下一层,保证 了while 循环在每一次迭代处理一层。而传统层序遍历是加入左右节点的,而这个题目里面我们是需要加入一个list的子节点,所以我们需要再使用一个循环把整个list的子节点入队。

代码

/*
// 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) {
       
        Queue<Node>queue=new LinkedList<>();
        queue.add(root);
        List<List<Integer>> res=new ArrayList<>(); if(root==null) return res;
        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 cur:node.children)
                    queue.add(cur);
                
            }
            res.add(list);
        }
        return res;
    }
}
  • 时间复杂度:O(n)。n 指的是节点的数量。
  • 空间复杂度:O(n)。