面试题 04.03. 特定深度节点链表

最新推荐文章于 2024-09-26 00:04:07 发布

胡矣

最新推荐文章于 2024-09-26 00:04:07 发布

阅读量317

点赞数 1

CC 4.0 BY-SA版权

分类专栏：算法 LeetCode 文章标签：链表 leetcode题解力扣题解二叉树 java

本文链接：https://blog.youkuaiyun.com/yirentianran/article/details/122003241

算法同时被 2 个专栏收录

142 篇文章

订阅专栏

LeetCode

130 篇文章

订阅专栏

本文介绍了一种算法，该算法可以将给定的二叉树转换为一系列链表，每个链表包含同一深度的所有节点。通过层序遍历的方式实现这一目标，并最终返回包含这些链表的数组。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

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

解题思路

层序遍历树中每层节点
每层节点生成一个链表

代码

/**
 * 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<>();
    Queue<TreeNode> queue = new LinkedList<>();
    queue.add(tree);
    while (!queue.isEmpty()) {
      int size = queue.size();
      ListNode lNode = new ListNode();
      for (int i = 0; i < size; i++) {
        TreeNode tNode = queue.poll();
        lNode.next = new ListNode(tNode.val);
        lNode = lNode.next;
        // 左子树
        if (tNode.left != null) queue.offer(tNode.left);
        // 右子树
        if (tNode.right != null) queue.offer(tNode.right);
        // 将头结点放入list中
        if (i == 0) list.add(lNode);
      }
    }
    return list.toArray(new ListNode[0]);
  }
}