Leetcode 429.N叉树的层次遍历 N-ary Tree Level Order Traversal(Java)

Leetcode 429.N叉树的层次遍历 N-ary Tree Level Order Traversal(Java)

##Tree##, ##Breadth-first Search##

N叉树的层序遍历

本题采用层序遍历BFS，借助队列实现

• 队列q中保存树的一层的所有结点
• 结果集合要求按层返回，每次拓展时，保证完整地扩展下一层的所有结点
• 获取q的长度len，len层循环，将该层的所有结点遍历加入结果集，并且将遍历结点的左右儿子结点加入q种

时间复杂度： O(n)

class Solution {
    public List<List<Integer>> levelOrder(Node root) {
        List<List<Integer>> res = new ArrayList<>();
        if (root == null) return res;
        Queue<Node> q = new LinkedList<>();
        q.offer(root);
        
        while (!q.isEmpty()) {
            int len = q.size();
            ArrayList<Integer> level = new ArrayList<>();
            for (int i = 0; i < len; i ++) {
                Node temp = q.poll();
                level.add(temp.val);
                for (Node n : temp.children) q.add(n);
            }
            res.add(level);
        }
        return res;
    }
}