leetcode-112-路径总和

题目描述

给你二叉树的根节点 root 和一个表示目标和的整数 targetSum 。判断该树中是否存在 根节点到叶子节点 的路径，这条路径上所有节点值相加等于目标和 targetSum 。如果存在，返回 true ；否则，返回 false 。

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

实例1：
输入：root = [5,4,8,11,null,13,4,7,2,null,null,null,1], targetSum = 22
输出：true
解释：等于目标和的根节点到叶节点路径如上图所示。

提示

树中节点的数目在范围 [0, 5000] 内
-1000 <= Node.val <= 1000
-1000 <= targetSum <= 1000

思路

对于二叉树，大都和遍历有关。
本题中关键是2点：

• 计算节点的值。
• 从根节点到叶子节点的路径。
• 返回值。

图、树等题型基本是遍历问题，可以从DFS遍历入手，计算节点值即相当于教材里对当前节点的访问，值得注意的是返回值问题，针对题目含义，存在一条这样的路径即可返回true。

解决

    # 采用 先序遍历 递归形式。
    def hasPathSum(self, root: TreeNode, targetSum: int) -> bool:
        def preOrder(root: TreeNode, current_sum: int):
            if root is not None:
                # 获取 当前节点 的值。
                current_sum += root.val
                # 当前 叶子节点。
                if root.left is None and root.right is None and current_sum == targetSum:
                    return True
                return preOrder(root.left, current_sum) or preOrder(root.right, current_sum)
            else:
                # 空树 返回false。
                return False

        return preOrder(root, 0)

返回值处理：

return preOrder(root.left, current_sum) or preOrder(root.right, current_sum)

结果