Java/429. N-ary Tree Level Order Traversal N叉树的层序遍历

最新推荐文章于 2022-04-08 09:31:52 发布

转载最新推荐文章于 2022-04-08 09:31:52 发布 · 915 阅读

文章标签：

# N-ary Tree Level Order Traver #N叉树的层序遍历

简单同时被 3 个专栏收录

117 篇文章

订阅专栏

Java

90 篇文章

订阅专栏

bfs

1 篇文章

订阅专栏

本文介绍两种实现二叉树层次遍历的算法，一种使用队列进行广度优先搜索，另一种采用递归方式进行宽度优先搜索。前者通过队列记录每一层节点，后者则直接在递归过程中记录各层次节点。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

代码部分一（9ms）

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

若树为空，返回 null
创建队列 queue ,并将根节点 root 入队，获得队列长度 size
当队列不为空时，取出(会删除)队首节点，将其值加入 list ,并将取出节点的子节点（下一层节点）入队
若当前层的节点是否都遍历完了，将新队列长度赋予size , 将 list 加入res，初始化 list
返回 res

代码部分二（5ms）

class Solution {
    List<List<Integer>> res = new ArrayList<List<Integer>>();
    public List<List<Integer>> levelOrder(Node root) {
        bfs(root, 0);
        return res;

    }
    public void bfs(Node root, int level){
        if(null == root)
            return;
        if(level == res.size())
            res.add(new ArrayList());
        res.get(level).add(root.val);
        for(Node node : root.children)
            bfs(node, level+1);
    }
}