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

总结

一打卡,就是一天!

你可能感兴趣的:(leetcode,算法,职场和发展,python,数据结构)