【二叉树】特定深度节点链表

0x00 题目

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

示例

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

        1
       /  \ 
      2    3
     / \    \ 
    4   5    7
   /
  8

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

---

0x01 思路

根据示例
是把二叉树的每一层都转换成一条链表
所以使用层序遍历方式
获取到一层数据后
再转换成链表即可

---

0x02 解法

语言：Swift

树节点：TreeNode

public class TreeNode {
    public var val: Int
    public var left: TreeNode?
    public var right: TreeNode?
    public init() { self.val = 0; self.left = nil; self.right = nil; }
    public init(_ val: Int) { self.val = val; self.left = nil; self.right = nil; }
    public init(_ val: Int, _ left: TreeNode?, _ right: TreeNode?) {
        self.val = val
        self.left = left
        self.right = right
    }
}

public class ListNode {
    public var val: Int
    public var next: ListNode?
    public init() { self.val = 0; self.next = nil; }
    public init(_ val: Int) { self.val = val; self.next = nil; }
    public init(_ val: Int, _ next: ListNode?) { self.val = val; self.next = next; }
}

解法：

func listOfDepth(_ tree: TreeNode?) -> [ListNode?] {
    guard let tree = tree else { return [] }
    
    var res: [ListNode?] = []
    var queue: [TreeNode?] = []
    queue.append(tree)
    
    while !queue.isEmpty {
        var temp: [TreeNode?] = []
        var arr: [Int] = []
        
        // 获取所有层的数值
        while !queue.isEmpty {
            let node = queue.removeFirst()
            arr.append(node!.val)
            
            if let left = node?.left {
                temp.append(left)
            }
            if let right = node?.right {
                temp.append(right)
            }
        }
        
        // 转换成链表
        if arr.count > 0 {
            let head = ListNode(arr[0])
            var curr = head
            for i in 1..<arr.count {
                let next = ListNode(arr[i])
                curr.next = next
                curr = next
            }
            res.append(head)
        }

        queue = temp
    }
    
    return res
}

---

0x03 我的小作品

欢迎体验我的作品之一：小编辑器-XCompiler
在线编辑器~小而巧
App Store 搜索即可~

---