路径总和
题目链接:力扣题目链接
难度:简单
给你二叉树的根节点 root 和一个表示目标和的整数 targetSum 。判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum 。如果存在,返回 true ;否则,返回 false 。
叶子节点 是指没有子节点的节点。
示例
输入:root = [5,4,8,11,null,13,4,7,2,null,null,null,1], targetSum = 22
输出:true
解释:等于目标和的根节点到叶节点路径如上图所示。
定义一个二叉树节点
//二叉树节点的定义。
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;
}
}
思路
假定从根节点到当前节点的值之和为val,我们可以将这个大问题转化为一个小问题:是否存在从当前节点到叶子的路径,满足其路径和为sum-val。
使用广度优先搜索,记录从根节点到当前节点的路径和,以防止重复计算。使用两个队列,分别存储将要遍历的节点,以及根节点到这些节点的路径和。
递归代码
class Solution{
public boolean hasPathSum(TreeNode root,int targetSum){
if(root == null) return false;
//当找到叶子节点时,如果剩下的targetSum值刚好和当前节点值相等就返回true
if(root.left == null && root.right == null) return root.val == targetSum;
//分别从左子树和右子树去找叶子节点 每次减去走过的节点值
return hasPathSum(root.left,targetSum - root.val) || hasPathSum(root.right,targetSum - root.val);
}
}
迭代代码
class Solution{
public boolean hasPathSum(TreeNode root,int targerSum){
//判断根节点存在不存在,不存在返回false
if(root == null) return false;
//创建两个队列分别存放节点和节点值
Queue<TreeNode> queueNode = new LinkedList<>();
Queue<Integer> queueVal = new LinkedList<>();
queueNode.offer(root);
queueVal.offer(root.val);
while(!queueNode.isEmpty()){
TreeNode node = queueNode.poll();
Integer value = queueVal.poll();
//如果取出的节点不存在子节点,那么就判断路径和与目标值是否相等
if(node.left == null && node.right == null){
if(value == targerSum){
return true;
}
}
//判断有左子节点的话
if(node.left != null){
//将左子节点加入第一个队列
queueNode.offer(node.left);
//将当前左子节点的值与路径和相加加入第二个队列
queueVal.offer(node.left.val + value);
}
if(node.right != null){
queueNode.offer(node.right);
queueVal.offer(node.right.val + value);
}
}
return false;
}
}