【坚持每日一题10.8】特定深度节点链表

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

示例：

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

    1
   /  \ 
  2    3
 / \    \ 
4   5    7

/
8

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/list-of-depth-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

java代码：

/**
 * 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) {
        List<ListNode> list = new ArrayList<>();
        dfs(list, tree, 1);
        ListNode[] arr = new ListNode[list.size()];
        for (int i = 0; i < arr.length; i++) {
            arr[i] = list.get(i);
        }
        return arr;
    }

    private void dfs(List<ListNode> list, TreeNode node, int deep) {
        if (node == null) {
            return;
        }
        if (deep > list.size()) {
            list.add(new ListNode(node.val));
        } else {
            ListNode n = list.get(deep - 1);
            while (n.next != null) {
                n = n.next;
            }
            n.next = new ListNode(node.val);
        }
        dfs(list, node.left, deep + 1);
        dfs(list, node.right, deep + 1);
    }
}

标签：node,ListNode,val,int,10.8,list,链表,TreeNode,一题
来源： https://blog.csdn.net/kangbin825/article/details/120661736