leetcode第112题,二叉树搜索问题,计算从根到叶子节点路径之和与目标数比较。
首先我们确定一下递归的出口条件,出口应该是叶子节点处,这个时候应该计算整个路径上的和了,并且与目标值比较,如果正确就返回True。如果不是叶子节点,那么先加上本节点的值,然后分别向左右子树递归,左右子树会出现两个值,分别代表这两条路径上是否具有满足条件的情况,因此最后或以下就可以了。
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def hasPathSum(self, root, sum):
def dfs(node,sum,target):
if node.left == None and node.right == None:
sum += node.val
if sum == target:
return True
else:
return False
else:
sum += node.val
l = r = False
if node.left != None:
l = dfs(node.left,sum,target)
if node.right != None:
r = dfs(node.right,sum,target)
return l or r
if root == None:
return False
return dfs(root,0,sum)