面试题 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]]

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/list-of-depth-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

 

复制代码
# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

# Definition for singly-linked list.
# class ListNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution(object):
    def listOfDepth(self, root):
        """
        :type tree: TreeNode
        :rtype: List[ListNode]
        """
        if root is None:
            return []
        res=[]
        #层次遍历
        def add_to_res(level,node):
            if level>len(res)-1:
                res.append([])
            res[level].append(node.val)
            if node.left:
                add_to_res(level+1,node.left)
            if node.right:
                add_to_res(level+1,node.right)

        add_to_res(0,root)

        def insert(root, item): 
            temp = ListNode(item) 
            #尾插法
            if (root == None): 
                root = temp 
            else : 
                ptr = root 
                while (ptr.next != None): 
                    ptr = ptr.next
                ptr.next = temp 
            
            return root 
        
        #array转linked list
        def arrayToList(arr, n): 
            root = None
            for i in range(0, n, 1): 
                root = insert(root, arr[i]) 
            
            return root 

        ans=[]

        for arr in res:
            ans.append(arrayToList(list,len(arr)))
        
        return ans

    
复制代码

 

posted @   XXXSANS  阅读(98)  评论(0编辑  收藏  举报
编辑推荐:
· 智能桌面机器人:用.NET IoT库控制舵机并多方法播放表情
· Linux glibc自带哈希表的用例及性能测试
· 深入理解 Mybatis 分库分表执行原理
· 如何打造一个高并发系统?
· .NET Core GC压缩(compact_phase)底层原理浅谈
阅读排行:
· 新年开篇:在本地部署DeepSeek大模型实现联网增强的AI应用
· DeepSeek火爆全网,官网宕机?本地部署一个随便玩「LLM探索」
· Janus Pro:DeepSeek 开源革新,多模态 AI 的未来
· 互联网不景气了那就玩玩嵌入式吧,用纯.NET开发并制作一个智能桌面机器人(三):用.NET IoT库
· 上周热点回顾(1.20-1.26)
点击右上角即可分享
微信分享提示