二叉树中和为某一值的路径

224 字

1 分钟

二叉树中和为某一值的路径

2021-09-06

2026-10-12

数据结构与算法

「剑指Offer」Swift

题目#

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

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

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

返回:

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

提示：

节点总数 <= 10000

来源：力扣（LeetCode）

题解#

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

1
func pathSum(_ root: TreeNode?, _ target: Int) -> [[Int]] {
2
    guard let rootV = root?.val else {
3
        return []
4
    }
5
    var result: [[Int]] = []
6
    let deta = target - rootV
7
    var temp: [TreeNode] = []
8

9
    if let left = root?.left {
10
        temp.append(left)
11
    }
12

13
    if let right = root?.right {
14
        temp.append(right)
15
    }
16

17
    if temp.isEmpty { // 到达叶节点了
18
        if target == rootV {
19
            result.append([rootV])
20
        }
21
    }
22
    else {
23
        temp.forEach { tree in
24
            pathSum(tree, deta).forEach { datas in
25
                result.append([rootV] + datas)
26
            }
27
        }
28
    }
29

30
    return result
31
}