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

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

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

示例:

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

    1
   /  \ 
  2    3
 / \    \ 
4   5    7

/
8

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

AC 代码:

/**
 * 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) {
         LinkedList<TreeNode> queue = new LinkedList<>();
         queue.offer(tree);
         List<ListNode> res = new ArrayList<ListNode>();
         ListNode list = new ListNode(-1);
         while(!queue.isEmpty()){
             int size = queue.size();
             ListNode cur = list;
             for(int i = 0;i < size; i++){
                 TreeNode node = queue.poll();
                 if(node.left != null){
                     queue.offer(node.left);
                 }

                 if(node.right != null){
                     queue.offer(node.right);
                 }
                 cur.next = new ListNode(node.val);
                 cur = cur.next; 
             }
             res.add(list.next);
             list.next = null;
         }

         return res.toArray(new ListNode[]{});

        

    }
}

欢迎关注公众号:程序员开发者社区

在这里插入图片描述

wangxiaoming CSDN认证博客专家 架构 Spring Boot Redis
博客是很好的总结和记录工具,如果有问题,来不及回复,关注微信公众号:程序员开发者社区,获取我的联系方式,向我提问,也可以给我发送邮件,联系 1275801617@qq.com
已标记关键词 清除标记
©️2020 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页
实付 19.89元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值