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