给定一棵二叉树,设计一个算法,创建含有某一深度上所有节点的链表(比如,若一棵树的深度为 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);
    }
}