题目来源:https://leetcode-cn.com/problems/path-sum-iii/description/
题目描述:
算法描述:该题的难度在上篇博客题目的基础上加大了一些难度。
详情请见: https://blog.csdn.net/qq_39241239/article/details/82777537。
相信你已经看了之前的博客了,那么这题就变得比较简单了。首先从根节点开始遍历二叉树,找出从根节点开始的符合题目条件的路径之后,把路径之和加1。当你遍历完以根节点为起点的二叉树之后。那么你就该遍历以根节点的左孩子为起点的二叉树。如此递归就能得到结果。
代码如下:
int depth(struct TreeNode* root, int sum,int *count){
if(root==NULL){
return 0;
}
//当遍历到一个节点时,发下从起点到该结点的路径符合要求,那么个数加1.
if(sum==root->val){
*count+=1;
}
//开始递归遍历左子树
depth(root->left,sum-root->val,count);
//开始递归遍历右子树
depth(root->right,sum-root->val,count);
return 0;
}
int pathSum(struct TreeNode* root, int sum) {
if(root==NULL){
return 0;
}
//因为要改变count的数组,所以传的参数为一个指针
int *count =(int*)malloc(sizeof(int));
*count=0;
//该函数执行完之后,*count的数值就是以根节点为起点的二叉树中符合条件的路径数之和。
depth(root,sum,count);
//然后再递归遍历左子树,右子树等等。
return *count+=pathSum(root->left,sum)+pathSum(root->right,sum);
}