【每日一题】LeetCode 112.路径总和(树、深度优先搜索、广度优先搜索、二叉树)
题目描述
给定一个二叉树的根节点 root
和一个整数 targetSum
,判断树中是否存在从根节点到叶子节点的路径,使得这条路径上所有节点的值加起来等于 targetSum
。如果存在,返回 true
;否则,返回 false
。
思路分析
这个问题可以通过深度优先搜索(DFS)来解决。我们从根节点开始,递归地遍历整棵树。对于每个节点,我们更新当前路径的和(即 targetSum
减去当前节点的值),然后继续递归左右子节点。如果当前节点是叶子节点(即没有子节点),我们检查当前路径的和是否等于目标和。如果等于,返回 true
;否则,继续递归。
关键点在于,我们需要在递归调用中传递当前节点值与目标和的差值,而不是使用一个全局变量来存储当前路径的和。这样可以避免在处理多个可能的路径时出现错误。
输入示例
-
示例 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
- 解释:不存在任何一条路径,其节点值之和等于目标和。
- 输入:
-
示例 3:
- 输入:
root = [], targetSum = 0
- 输出:
false
- 解释:树为空,不存在任何路径。
- 输入:
代码实现
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public boolean hasPathSum(TreeNode root, int targetSum) {
return dfs(root, targetSum);
}
/**
* 深度优先搜索(DFS)函数,用于递归检查路径和。
* @param node 当前节点
* @param targetSum 目标和
* @return 如果当前路径和等于目标和,返回true;否则,返回false。
*/
public boolean dfs(TreeNode node, int targetSum) {
if (node == null) {
// 如果节点为空,说明路径不成立
return false;
}
// 如果是叶子节点,检查当前节点的值是否等于目标和
if (node.left == null && node.right == null) {
return targetSum == node.val;
}
// 递归检查左右子树,更新目标和为当前节点值与目标和的差
return dfs(node.left, targetSum - node.val) || dfs(node.right, targetSum - node.val);
}
}