给你二叉树的根节点 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 的根节点到叶子节点的路径。
示例 3:
输入:root = [], targetSum = 0 输出:false 解释:由于树是空的,所以不存在根节点到叶子节点的路径。
提示:
- 树中节点的数目在范围
[0, 5000]
内 -1000 <= Node.val <= 1000
-1000 <= targetSum <= 1000
代码如下:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* };
*/
int search(struct TreeNode* root,int sum,int targetSum,bool *flag){
if(*flag){// 已经找到,后面不用计算
return sum;
}
sum += root->val;
//在这里进行绝对值翻转,然后比对sum和target的值,如果sum的绝对值已经比target大可以不用在进行递归,直接在这里返回结束
if(!root->left&&!root->right){// 当节点为叶子节点时进行比对
if(sum == targetSum) *flag = true;
}
// 进行深度递归
if(root->left){
search(root->left,sum,targetSum,flag);
}
if(root->right){
search(root->right,sum,targetSum,flag);
}
return sum;
}
bool hasPathSum(struct TreeNode* root, int targetSum) {
if(!root) return false;// 空树
bool flag = false;
if(root){
search(root,0,targetSum,&flag);
}
return flag;
}
总得来说,当代码提交通过的时候再去看这一道题,其实也不是很难。只要使用sum计算出某条路径的总和即可。可优化的点有:1.当某条路径的值已经大于目标值时可以不用再去计算后续节点。2.已经找到了一条路径,则后面的路径可以不用再进行计算。
ps:如有更好的方案,请各位大佬指教OVO。。