LeetCode112 路徑總和 2018.11.13
阿新 • • 發佈:2018-12-21
題幹:
給定一個二叉樹和一個目標和,判斷該樹中是否存在根節點到葉子節點的路徑,這條路徑上所有節點值相加等於目標和。
說明: 葉子節點是指沒有子節點的節點。
示例:
給定如下二叉樹,以及目標和 sum = 22
,
5 / \ 4 8 / / \ 11 13 4 / \ \ 7 2 1
返回 true
, 因為存在目標和為 22 的根節點到葉子節點的路徑 5->4->11->2
。
class Solution: def hasPathSum(self, root, sum): """ :type root: TreeNode :type sum: int :rtype: bool """ if not root: return False elif not root.left and not root.right: if root.val == sum: return True else: return False else: sum -= root.val return self.hasPathSum(root.left, sum) or self.hasPathSum(root.right, sum)