使用按层BFS来解决“N叉树的层序遍历”的问题

Posted 2022-04-12 &小小白&

六、N叉树的层序遍历

6.1、题设要求

​  给定一个 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] 之间

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

6.2、解题思路

​  我们的做法就是按层BFS，将当前层所有的节点加到结果里，然后将他们的子节点加到下一层，遍历完了之后用下一层替换当前层进行遍历。

6.3、算法

class Solution 
    public List<List<Integer>> levelOrder(Node root) 
        List<List<Integer>> ans = new ArrayList<>();
        Queue<Node> queue = new ArrayDeque<>();

        //如果root为null,输出空
        if(root == null)
            return ans;
        

        queue.offer(root);
        while(!queue.isEmpty())
            //当前层的大小
            int cnt = queue.size();
            //每层定义一个新容器
            List<Integer> level = new ArrayList<>();
            //对当前层进行遍历
            for(int i = 0;i < cnt;++i)
                //将当前节点添加到level中
                Node cur = queue.poll();
                level.add(cur.val);
                //添加ans结果集到下一层
                for(Node child : cur.children)
                    queue.offer(child);
                
            
            ans.add(level);
        
        return ans;