DFS回溯只在递归基回溯————leetcode112

最新推荐文章于 2024-08-22 11:48:29 发布

weixin_30888413

最新推荐文章于 2024-08-22 11:48:29 发布

阅读量62

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/vector11248/p/9712554.html

本文探讨了二叉树中是否存在从根节点到叶子节点的路径，使得沿途经过的节点值之和等于给定的目标值。通过深度优先搜索(DFS)算法实现，递归遍历所有可能的路径并计算其总和。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

# class Solution:
#     def hasPathSum(self, root, sum):
#         """
#         :type root: TreeNode
#         :type sum: int
#         :rtype: bool
#         """
class Solution:
    
    res = False
    
    def hasPathSum(self, root, sum):
        """
        :type root: TreeNode
        :type sum: int
        :rtype: bool
        """
        if root is None:
            return False
        self.dfs2(root,0,sum)
        return self.res

    

    def dfs2(self, node, temp_sum, sum):
        temp_sum += node.val
        if not node.left and not node.right:
            if temp_sum == sum:
                self.res = True
                return
            temp_sum -= node.val
            return
        if node.left:
            self.dfs2(node.left,temp_sum,sum)
        if node.right:
            self.dfs2(node.right,temp_sum,sum)