[算法练习及思路-程序员面试金典(Java解法)]No26.特定深度节点链表

题号:no26

题目名:特定深度节点链表

原题URL:https://leetcode-cn.com/problems/list-of-depth-lcci/

题目描述

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

示例

示例 1：

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

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

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

思路

1.使用层序遍历,模板通用

2.在每次遍历的时候新建一个ListNode数组

3.每次将tree的结点加到ListNode的后边

4.添加到集合中然后进行转换

解题代码

public class Solution {
    public ListNode[] listOfDepth(TreeNode tree) {
        //层序遍历，将每层的结点结合在一个链表中进行存储
        if(tree==null) return null;
        List<ListNode> resList = new ArrayList<>();
        ArrayDeque<TreeNode> queue = new ArrayDeque<>();
        queue.add(tree);
        //层序遍历,基本模板化,在层序遍历的每一层 需要将所有层的结点拿到,放入到集合中
        while (!queue.isEmpty()) {
            int size = queue.size();
            ListNode dummy = new ListNode(0);//虚拟头结点
            ListNode node = dummy;
            while(size!=0) {
                TreeNode treeNode = queue.removeFirst();
                if(treeNode.left!=null) queue.add(treeNode.left);
                if(treeNode.right!=null) queue.add(treeNode.right);
                size--;
                node.next = new ListNode(treeNode.val);
                node = node.next;
            }
            resList.add(dummy.next);
        }
        return resList.toArray(new ListNode[0]);
    }
}