【剑指Offer-Swift】34.二叉树中和为某一值的路径

223 阅读1分钟

「这是我参与11月更文挑战的第2天,活动详情查看:2021最后一次更文挑战

Hi 👋

我的个人项目扫雷Elic 无尽天梯梦见账本隐私访问记录
类型游戏财务工具
AppStoreElicUmemi隐私访问记录

更多专栏:

Lawliet的独立开发碎碎念

Lawliet的iOS游园会

Lawliet的iOS底层实验室

Lawliet的iOS逆向实验室

Lawliet的刷题小本本

题目

输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。

示例: 给定如下二叉树,以及目标和 target = 22,

               5
              / \
             4   8
            /   / \
           11  13  4
          /  \    / \
         7    2  5   1

返回:

 [
    [5,4,11,2],
    [5,8,4,5]
 ]

提示:

节点总数 <= 10000

叶子节点 是指没有子节点的节点。

来源:力扣(LeetCode)

题解

将问题拆解为子树的问题,递归解决

func pathSum(_ root: TreeNode?, _ target: Int) -> [[Int]] {
    guard let rootV = root?.val else {
        return []
    }
    var result: [[Int]] = []
    let deta = target - rootV
    var temp: [TreeNode] = []
    
    if let left = root?.left {
        temp.append(left)
    }
    
    if let right = root?.right {
        temp.append(right)
    }

    if temp.isEmpty { // 到达叶节点了
        if target == rootV {
            result.append([rootV])
        }
    }
    else {
        temp.forEach { tree in
            pathSum(tree, deta).forEach { datas in
                result.append([rootV] + datas)
            }
        }
    }
    
    return result
}

34-01.png

👋欢迎点赞收藏关注♥️