zl程序教程

您现在的位置是:首页 >  其他

当前栏目

LeetCode_多叉树_中等_429.N 叉树的层序遍历

LeetCode遍历 中等 层序
2023-09-27 14:25:45 时间

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]]

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/n-ary-tree-level-order-traversal

2.思路

(1)层序遍历
本题与102.二叉树的层序遍历这题十分相似,只不过遍历的对象由二叉树变为了多叉树,但是核心的遍历思想还是一样的,即通过 while 循环控制从上向下方向的遍历,以及 for 循环控制每一层从左向右的遍历

3.代码实现(Java)

//思路1————层序遍历
/*
// 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) {
        // res 用于保存最终的结果
        List<List<Integer>> res = new ArrayList<>();
        if (root == null) {
            return res;
        }
        // queue 中保存层序遍历过程中每一层的所有节点
        Queue<Node> queue = new LinkedList<>();
        // 第一层只有一个根节点,先将其加入到 queue 中
        queue.offer(root);
        // while 循环控制从上向下方向的遍历
        while (!queue.isEmpty()) {
            // levelSize 记录当前层的节点个数
            int levelSize = queue.size();
            // levelVals 保存当前层的所有节点值
            List<Integer> levelVals = new ArrayList<>();
            // for 循环控制每一层从左向右的遍历
            for (int i = 0; i < levelSize; i++) {
                Node curNode = queue.poll();
                levelVals.add(curNode.val);
                // 将当前的节点的子节点保存到 queue 中
                for (Node child : curNode.children) {
                    queue.offer(child);
                }
            }
            // 将当前层的所有节点值加入到 res 中
            res.add(levelVals);
        }
        return res;
    }
}