【二叉树】二叉树路径总和 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 搜索即可