【第九周】特定深度节点链表

LeetCode 面试题 04.03. 特定深度节点链表

广度优先遍历

按层遍历，在循环时首先得到队列中的元素个数，此个数即为树中每一层的元素个数，依次取出当前层的元素插入到链表中。

class Solution {
    public ListNode[] listOfDepth(TreeNode tree) {
        if (tree == null) return new ListNode[0];
        List list = new ArrayList<>();
        Queue queue = new LinkedList<>();
        queue.offer(tree);
        while (!queue.isEmpty()) {
            int size = queue.size();
            ListNode head = new ListNode(0);
            ListNode cur = head;
            for (int i = 0; i < size; i++) {
                TreeNode node = queue.poll();
                cur.next = new ListNode(node.val);
                cur = cur.next;
                if (node.left != null) queue.offer(node.left);
                if (node.right != null) queue.offer(node.right);
            }
            list.add(head.next);
        }
        return list.toArray(new ListNode[list.size()]);
    }
}