Leetcode——112. 路徑總和

大寫的ZDQ發表於2019-03-18

給定一個二叉樹和一個目標和,判斷該樹中是否存在根節點到葉子節點的路徑,這條路徑上所有節點值相加等於目標和。

說明: 葉子節點是指沒有子節點的節點。

示例:
給定如下二叉樹,以及目標和 sum = 22,

          5
         / \
        4   8
       /   / \
      11  13  4
     /  \      \
    7    2      1

返回 true, 因為存在目標和為 22 的根節點到葉子節點的路徑 5->4->11->2。

思路:遞迴,需要注意遞迴的終止點,如果左右子樹都為None,直接返回,然後遞迴。

class Solution(object):
    def hasPathSum(self, root, sum):
        """
        :type root: TreeNode
        :type sum: int
        :rtype: bool
        """
        if root == None:
            return False
        if root.left==None and root.right==None:
            return  root.val==sum
        if (self.hasPathSum(root.left,sum-root.val)):
            return True
        if (self.hasPathSum(root.right,sum-root.val)):
            return True
        return False

相關文章