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

给定一棵二叉树，设计一个算法，创建含有某一深度上所有节点的链表（比如，若一棵树的深度为 D，则会创建出 D 个链表）。返回一个包含所有深度的链表的数组。

示例：

输入：[1,2,3,4,5,null,7,8]

        1
       /  \ 
      2    3
     / \    \ 
    4   5    7
   /
  8

输出：[[1],[2,3],[4,5,7],[8]]

思路：根据题意，创建同一深度上所有结点的链表，所以dfs和bfs中，bfs更适合本题，确定了遍历方式后，思考如何建立同一层的链表，广度优先搜索遍历需要用到队列（FIFO），可以使结点的左右子节点先后进入队列中，以queue.size()为终止条件，进行遍历，即可将左右子节点（同一层结点）加入链表。同时将下一层加入队列中。

解法：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
   
    public ListNode[] listOfDepth(TreeNode tree) {
        //广度优先搜索
        if(tree == null) return null;
        List<ListNode> res = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(tree);
        while(!queue.isEmpty()){
            int queueSize = queue.size();
            ListNode node = new ListNode(0);
            ListNode head = node;
            for(int i = 0; i < queueSize; i++){
                TreeNode root = queue.poll();
                ListNode n = new ListNode(root.val);
                node.next = n;
                node = n;
                if(root.left != null){
                    queue.add(root.left);
                }
                if(root.right != null){
                    queue.add(root.right);
                }  
            }
            res.add(head.next);
        }
        return res.toArray(new ListNode[res.size()]);
    }
}