leetcode 面试题 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]]

 

 

 

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