给你二叉树的根节点 root 和一个表示目标和的整数 targetSum ,判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum 。
叶子节点 是指没有子节点的节点。
示例 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 = [1,2], targetSum = 0
输出:false
1.自己的解法:利用递归。自己写的这段代码每一段路径都需要计算到末尾,其实不需要这样使用,可以进行改进。
/**
* 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) {
int currentSum = 0;//当前路径和
return hasPathSum(root,targetSum,currentSum);//利用方法的重载
}
public boolean hasPathSum(TreeNode node,int targetSum,int currentSum){
if(node == null){//遍历完一条路径,都没有找到,说明返回false
return false;
}
currentSum += node.val;//当前节点存在时,更新当前路径和
if(currentSum == targetSum && node.left == null && node.right == null){//如果路径和满足要求且当前节点是叶子节点
return true;
}
return hasPathSum(node.left,targetSum,currentSum) || hasPathSum(node.right,targetSum,currentSum);//继续寻找
}
}
2.答案解法:
1.递归。
class Solution {
public boolean hasPathSum(TreeNode root, int sum) {
if (root == null) {//节点不存在,返回false
return false;
}
if (root.left == null && root.right == null) {//当前节点是叶子节点
return sum == root.val;//剩余的路径和是否等于当前节点的值
}
return hasPathSum(root.left, sum - root.val) || hasPathSum(root.right, sum - root.val);//继续向下遍历,路径和要减去当前节点的值
}
}
2.广度优先遍历。利用两个队列来分别存储将要遍历的节点以及根节点到这些节点的路径和即
可。
class Solution {
public boolean hasPathSum(TreeNode root, int sum) {
if(root == null){//树为空,返回false
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();//当前节点
int currentSum = queueVal.poll();//当前值
if(node.left == null && node.right == null){//当前节点是叶子节点
if(currentSum == sum){//如果当前和等于目标和,返回true,否则继续下一次循环
return true;
}
continue;
}
if(node.left != null){//左子节点存在,则加入队列
queueNode.offer(node.left);
queueVal.offer(currentSum + node.left.val);
}
if(node.right != null){//右子节点存在,则加入队列
queueNode.offer(node.right);
queueVal.offer(currentSum + node.right.val);
}
}
return false;
}
}
题源:力扣