【二叉树】二叉树路径总和 II

0x00 题目

给你二叉树的根节点 root
和一个整数目标和 targetSum
找出 所有 从 根节点 到 叶子节点 路径总和
等于给定 目标和 的路径

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

---

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
    }
}

解法：

func pathSum(_ root: TreeNode?, _ targetSum: Int) -> [[Int]] {
    guard let root = root else { return [] }
    
    // 存放结果
    var res: [[Int]] = []
    // 存放节点值
    var arr: [Int] = []

    func dfs(_ root: TreeNode?, _ target: Int) {
        guard let root = root else { return }
        // 添加一个数
        arr.append(root.val)
        
        // 减小目标
        let val = target - root.val
        
        // 叶子节点
        if root.left == nil && root.right == nil {
            // 符合条件
            if val == 0 { 
                res.append(arr)
            }
            // 删除最后一个数，返回后，添加其他节点值继续判断
            arr.removeLast() 
            return
        }

        dfs(root.left, val)
        dfs(root.right, val)
        
        // 删除最后一个数，返回后，添加其他节点值继续判断
        arr.removeLast() 
    }

    dfs(root, targetSum)

    return res
}

---

小笔记

做笔记，一步到位
App Store 搜索即可

---