路徑總和

題目

給定一個(gè)二叉樹(shù)和一個(gè)目標(biāo)和，判斷該樹(shù)中是否存在根節(jié)點(diǎn)到葉子節(jié)點(diǎn)的路徑，這條路徑上所有節(jié)點(diǎn)值相加等于目標(biāo)和。

說(shuō)明: 葉子節(jié)點(diǎn)是指沒(méi)有子節(jié)點(diǎn)的節(jié)點(diǎn)。

示例:
給定如下二叉樹(shù)，以及目標(biāo)和 sum = 22，

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

返回 true, 因?yàn)榇嬖谀繕?biāo)和為 22 的根節(jié)點(diǎn)到葉子節(jié)點(diǎn)的路徑 5->4->11->2。

解答

• 思路：

  • 遞歸進(jìn)行先序遍歷；
  • 如果遍歷到某個(gè)葉子節(jié)點(diǎn)，發(fā)現(xiàn)路徑總和為目標(biāo)值，則返回True。

• 代碼：

  def hasPathSum(self, root, sum):
      """
      :type root: TreeNode
      :type sum: int
      :rtype bool
  
      (knowledge)
  
      思路：
      1. 遞歸進(jìn)行先序遍歷；
      2. 如果遍歷到某個(gè)葉子節(jié)點(diǎn)，發(fā)現(xiàn)路徑總和為目標(biāo)值，則返回True
      """
      if not root:
          return False
  
      # 找到葉子節(jié)點(diǎn)進(jìn)行判斷
      if not root.left and not root.right and root.val == sum:
          return True
  
      return self.hasPathSum(root.left, sum - root.val) or self.hasPathSum(root.right, sum - root.val)
  

測(cè)試驗(yàn)證

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

        (knowledge)

        思路：
        1. 遞歸進(jìn)行先序遍歷；
        2. 如果遍歷到某個(gè)葉子節(jié)點(diǎn)，發(fā)現(xiàn)路徑總和為目標(biāo)值，則返回True
        """
        if not root:
            return False

        # 找到葉子節(jié)點(diǎn)進(jìn)行判斷
        if not root.left and not root.right and root.val == sum:
            return True

        return self.hasPathSum(root.left, sum - root.val) or self.hasPathSum(root.right, sum - root.val)


if __name__ == '__main__':
    solution = Solution()
    root = TreeNode(5)
    root.left = TreeNode(4)
    root.left.left = TreeNode(11)
    root.left.left.left = TreeNode(7)
    root.left.left.right = TreeNode(2)

    root.right = TreeNode(8)
    root.right.left = TreeNode(13)
    root.right.right = TreeNode(4)
    root.right.right.right = TreeNode(1)
    print(solution.hasPathSum(root, 22), "= True")