精品推荐:
《征服数据结构》专栏:50多种数据结构彻底征服。
《经典图论算法》专栏:50多种经典图论算法全部掌握。
一网友说公司部门裁员,把一个月薪一万五和一个月薪三万的裁掉了,并且把他们的工作都交给了该网友,该网友提出涨薪变成十恶不赦了。
三个人的活让一个人干,这谁能抗的住,如果真想让一个人干涨点工资也很正常,但是没通过,所有网友都劝他赶紧跑。
--------------下面是今天的算法题--------------
来看下今天的算法题,这题是LeetCode的第112题:路径总和。
问题描述
来源:LeetCode第112题
难度:简单
给你二叉树的根节点 root 和一个表示目标和的整数 targetSum 。判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和 targetSum 。如果存在,返回 true ;否则,返回 false 。
示例1:
输入:root = [5,4,8,11,null,13,4,7,2,null,null,null,1], targetSum = 22
输出:true
解释:等于目标和的根节点到叶节点路径如上图所示。
示例2:
输入:root = [1,2,3], targetSum = 5
输出:false
解释:树中存在两条根节点到叶子节点的路径:
(1 --> 2): 和为 3
(1 --> 3): 和为 4
不存在 sum = 5 的根节点到叶子节点的路径。
树中节点的数目在范围 [0, 5000] 内
-1000 <= Node.val <= 1000
-1000 <= targetSum <= 1000
问题分析
这题让判断是否存在从根节点到叶子节点的值等于targetSum的路径,我们从根节点开始用targetSum减去路径上节点的值,如果到叶子节点的时候,剩余的值等于叶子节点的值,说明存在这样的路径,直接返回true。
JAVA:
public boolean hasPathSum(TreeNode root, int targetSum) {
if (root == null)
return false;
// 如果到叶子节点了,并且剩余值等于叶子节点的值,返回true。
if (root.left == null && root.right == null && targetSum == root.val)
return true;
// 分别沿着左右子节点走下去,然后顺便把当前节点的值减掉,
// 左右子节点只要有一个返回true,说明存在这样的结果。
return hasPathSum(root.left, targetSum - root.val)
|| hasPathSum(root.right, targetSum - root.val);
}
C++:
public:
bool hasPathSum(TreeNode *root, int targetSum) {
if (root == nullptr)
return false;
// 如果到叶子节点了,并且剩余值等于叶子节点的值,返回true。
if (root->left == nullptr && root->right == nullptr && targetSum == root->val)
return true;
// 分别沿着左右子节点走下去,然后顺便把当前节点的值减掉,
// 左右子节点只要有一个返回true,说明存在这样的结果。
return hasPathSum(root->left, targetSum - root->val)
|| hasPathSum(root->right, targetSum - root->val);
}
Python:
def hasPathSum(self, root: Optional[TreeNode], targetSum: int) -> bool:
if not root:
return False
# 如果到叶子节点了,并且剩余值等于叶子节点的值,返回true。
if not root.left and not root.right:
return targetSum == root.val
# 分别沿着左右子节点走下去,然后顺便把当前节点的值减掉,
# 左右子节点只要有一个返回true,说明存在这样的结果。
return (self.hasPathSum(root.left, targetSum - root.val) or
self.hasPathSum(root.right, targetSum - root.val))
笔者简介
博哥,真名:王一博,毕业十多年,《算法秘籍》作者,专注于数据结构和算法的讲解,在全球30多个算法网站中累计做题2000多道,在公众号中写算法题解800多题,对算法题有自己独特的解题思路和解题技巧,喜欢的可以给个关注,也可以下载我整理的1000多页的PDF算法文档。