LeetCode:112(Python)—— 路径总和(简单)

路径总和

概述:给你二叉树的根节点 root 和一个表示目标和的整数 targetSum 。判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和 targetSum 。如果存在,返回 true ;否则,返回 false 。叶子节点是指没有子节点的节点。

输入:root = [5,4,8,11,null,13,4,7,2,null,null,null,1], targetSum = 22
输出:true

输入:root = [1,2,3], targetSum = 5
输出:false

方法一:递归

思路:若当前节点就是叶子节点,那么我们直接判断 sum 是否等于 val 即可。若当前节点不是叶子节点,我们只需要递归地询问它的子节点是否能满足条件即可。

# 递归
class Solution:
    def hasPathSum(self, root: Optional[TreeNode], targetSum: int) -> bool:
        if not root:
            return False
        if not root.left and not root.right:
            return root.val == targetSum
        return self.hasPathSum(root.left, targetSum - root.val) \
        or self.hasPathSum(root.right, targetSum - root.val)

方法二:广度优先搜索

思路:从根节点到当前节点的路径和,以防止重复计算。这样我们使用两个队列,分别存储将要遍历的节点,以及根节点到这些节点的路径和即可。

# 广度优先搜索
class Solution:
    def hasPathSum(self, root: Optional[TreeNode], targetSum: int) -> bool:
        if not root:
            return False
        queue_node = collections.deque([root])
        queue_val = collections.deque([root.val])
        while queue_node:
            now = queue_node.popleft()
            temp = queue_val.popleft()
            if not now.left and not now.right:
                if temp == targetSum:
                    return True
                continue
            if now.left:
                queue_node.append(now.left)
                queue_val.append(now.left.val + temp)
            if now.right:
                queue_node.append(now.right)
                queue_val.append(now.right.val + temp)
        return False

总结

一打卡,就是一天!

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值