leetcode_112. 路径总和

举报
悲恋花丶无心之人 发表于 2021/02/03 00:21:35 2021/02/03
【摘要】 目录 一、题目内容 二、解题思路 三、代码 一、题目内容 给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。 说明: 叶子节点是指没有子节点的节点。 示例:  给定如下二叉树,以及目标和 sum = 22,         ...

目录

一、题目内容

二、解题思路

三、代码


一、题目内容

给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。

说明: 叶子节点是指没有子节点的节点。

示例: 
给定如下二叉树,以及目标和 sum = 22,

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \      \
        7    2      1
返回 true, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2。

二、解题思路

前序遍历,如果到达叶子结点tmp的和为sum就返回True,否则返回False。

三、代码


  
  1. # Definition for a binary tree node.
  2. class TreeNode:
  3. def __init__(self, x):
  4. self.val = x
  5. self.left = None
  6. self.right = None
  7. def __repr__(self):
  8. return str(self.val)
  9. class Solution:
  10. def hasPathSum(self, root: TreeNode, sum_num: int) -> bool:
  11. if not root:
  12. return False
  13. self.sum = sum_num
  14. def dfs(root, tmp):
  15. if sum(tmp) == self.sum and root.left is None and root.right is None:
  16. return True
  17. if sum(tmp) != self.sum and root.left is None and root.right is None:
  18. return False
  19. if root.left:
  20. left = dfs(root.left, tmp + [root.left.val])
  21. else:
  22. left = False
  23. if root.right:
  24. right = dfs(root.right, tmp + [root.right.val])
  25. else:
  26. right = False
  27. return right or left
  28. return dfs(root, [root.val])
  29. if __name__ == '__main__':
  30. a = TreeNode(5)
  31. a.left = TreeNode(4)
  32. a.right = TreeNode(8)
  33. a.left.left = TreeNode(11)
  34. a.right.left = TreeNode(13)
  35. a.right.right = TreeNode(4)
  36. a.left.left.left = TreeNode(7)
  37. a.left.left.right = TreeNode(2)
  38. a.right.right.right = TreeNode(1)
  39. sum_num = 22
  40. s = Solution()
  41. ans = s.hasPathSum(a, sum_num)
  42. print(ans)

文章来源: nickhuang1996.blog.csdn.net,作者:悲恋花丶无心之人,版权归原作者所有,如需转载,请联系作者。

原文链接:nickhuang1996.blog.csdn.net/article/details/108810831

【版权声明】本文为华为云社区用户转载文章,如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。