给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22,
5 / 4 8 / / 11 13 4 / 7 2 1
返回 true, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2。
解法一:将二叉树的各路径求和,分别放入数组中,然后判定目标值是否在数组中
# Definition for a binary tree node.# class TreeNode:# def __init__(self, x):# self.val = x# self.left = None# self.right = Noneclass Solution: sums=[] def getSum(self,root:TreeNode,lin=0): if not root: return False lin += root.val if not root.left and not root.right: self.sums.append(lin) return False self.getSum(root.left, lin) self.getSum(root.right, lin) def hasPathSum(self, root: TreeNode, sum: int) -> bool: self.getSum(root) if sum in self.sums: return True else: return False
解法二:解法一的反向思维,每次循环递减,当最后的值与节点值相等,说明存在
# Definition for a binary tree node.# class TreeNode:# def __init__(self, x):# self.val = x# self.left = None# self.right = Noneclass Solution: def hasPathSum(self, root: TreeNode, sum: int) -> bool: if not root: return False print(sum,root.val) if not root.left and not root.right and sum==root.val: return True if self.hasPathSum(root.left,sum-root.val) or self.hasPathSum(root.right,sum-root.val): return True else: return False