#yyds干货盘点# LeetCode程序员面试金典：特定深度节点链表

Posted 2022-12-23

题目：

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

        public ListNode[] listOfDepth(TreeNode tree) 
            dfs(tree, 0);
            ListNode[] resArr = new ListNode[res.size()];
            for (int j = 0, resSize = res.size(); j < resSize; j++) 
                ArrayList<ListNode> deepList = res.get(j);
                for (int i = 0; i < deepList.size(); i++) 
                    if (i == 0) 
                        resArr[j] = deepList.get(i);
                     else 
                        deepList.get(i - 1).next = deepList.get(i);
                    
                
            
            return resArr;
        

        private void dfs(TreeNode tree, int index) 
            if (tree == null) return;
            if (res.size() <= index) 
                res.add(index, new ArrayList<>());
            
            ArrayList<ListNode> listNodes = res.get(index);
            listNodes.add(new ListNode(tree.val));
            dfs(tree.left, index + 1);
            dfs(tree.right, index + 1);