104 字 少于 1 分钟阅读

给定一个 N 叉树,返回其节点值的层序遍历。(即从左到右,逐层遍历)。

树的序列化输入是用层序遍历,每组子节点都由 null 值分隔(参见示例)。

示例 1:

img

输入:root = [1,null,3,2,4,null,5,6]
输出:[[1],[3,2,4],[5,6]]

示例 2:

img

输入: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] 之间

思路

102. 二叉树的层序遍历相同。

解法

class Solution {
    public List<List<Integer>> levelOrder(Node root) {
        List<List<Integer>> resList = new ArrayList<>();
        Deque<Node> deque = new LinkedList<>();
        if (root == null) {
            return resList;
        }
        deque.offer(root);
        int len = 0;
        while (!deque.isEmpty()) {
            len = deque.size();
            List<Integer> list = new ArrayList<>();
            while (len > 0 && !deque.isEmpty()) {
                root = deque.poll();
                list.add(root.val);
                if (root.children != null) {
                    for (Node node : root.children) {
                        deque.offer(node);
                    }
                }
                len--;
            }
            resList.add(list);
        }
        return resList;
    }
}