【二叉树】二叉树路径总和问题

0x00 题目

给你一棵二叉树的根节点 root 和一个表示目标 和 的整数 targetSum
判断该树中是否存在【根节点】到【叶子节点】的路径
这条路径上所有节点值相加的结果等于 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 hasPathSum(_ root: TreeNode?, _ targetSum: Int) -> Bool {
    // 无路可走了，返回
    guard let root = root else { return false }
    
    // 如果是叶子节点，判断跟 目标 是否相等
    if root.left == nil && root.right == nil {
        return root.val == targetSum
    }
    
    // 往下走，“步数” 减少
    let sum = targetSum - root.val
    return hasPathSum(root.left, sum) || hasPathSum(root.right, sum)
}

---

小编辑器

---