原题链接:二叉树中和为某一值的路径(一)_牛客题霸_牛客网
描述:
给定一个二叉树root和一个值 sum ,判断是否有从根节点到叶子节点的节点值之和等于 sum 的路径。
1.该题路径定义为从树的根结点开始往下一直到叶子结点所经过的结点
2.叶子节点是指没有子节点的节点
3.路径只能从父节点到子节点,不能从子节点到父节点
4.总节点数目为n
例如:
给出如下的二叉树,sum=22,
返回true,因为存在一条路径 5\to 4\to 11\to 25→4→11→2的节点值之和为 22
数据范围:
1.树上的节点数满足0≤n≤10000
2.每 个节点的值都满足val≤1000
要求:空间复杂度 O(n),时间复杂度 O(n)
进阶:空间复杂度 O(树的高度),时间复杂度 O(n)
解题思路:
二叉树的题基本上都是动态规划的解法。
遍历一个节点,如果这个节点left和right都为空,则判断是否等于给定值sum,等于则返回true,否则是false。
如果不为空,继续递归遍历其left和right节点,传入的sum=sum-当前节点的val。
代码:
/**
* @param root TreeNode类
* @param sum int整型
* @return bool布尔型
*/
public boolean hasPathSum(TreeNode root, int sum) {
if (root == null) {
return false;
}
if (root.left == null && root.right == null) {
return root.val == sum;
}
boolean isLeft = false;
boolean isRight = false;
if (root.left != null) {
isLeft = hasPathSum(root.left, sum - root.val);
}
if (root.right != null) {
isRight = hasPathSum(root.right, sum - root.val);
}
return isLeft || isRight;
}