当前位置 : 主页 > 编程语言 > java >

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

来源:互联网 收集:自由互联 发布时间:2023-02-04
题目: 给定一棵二叉树,设计一个算法,创建含有某一深度上所有节点的链表(比如,若一棵树的深度为 D,则会创建出 D 个链表)。返回一个包含所有深度的链表的数组。 示例: 输

题目:

给定一棵二叉树,设计一个算法,创建含有某一深度上所有节点的链表(比如,若一棵树的深度为 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); } }
上一篇:#yyds干货盘点# 名企真题专题: 棋子翻转
下一篇:没有了
网友评论