C++刷leetcode(面试题 04.12. 求和路径)【深度优先搜索】
本帖最后由 糖逗 于 2020-4-24 13:12 编辑题目描述:
给定一棵二叉树,其中每个节点都含有一个整数数值(该值或正或负)。设计一个算法,打印节点数值总和等于某个给定值的所有路径的数量。注意,路径不一定非得从二叉树的根节点或叶节点开始或结束,但是其方向必须向下(只能从父节点指向子节点方向)。
示例:
给定如下二叉树,以及目标和 sum = 22,
5
/ \
4 8
/ / \
11134
/\ / \
7 25 1
返回:
3
解释:和为 22 的路径有:, ,
提示:
节点总数 <= 10000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/paths-with-sum-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
int res = 0;
void solution(TreeNode* root, int total, int sum){
if(total == sum) res++;
if(root -> left != NULL)solution(root -> left, total + root -> left -> val, sum);
if(root -> right != NULL) solution(root -> right, total+root->right->val, sum);
}
void dfs(TreeNode* root, int sum){
if(root == NULL) return;
solution(root, root -> val, sum);
dfs(root -> left, sum);
dfs(root -> right, sum);
}
int pathSum(TreeNode* root, int sum) {
dfs(root, sum);
return res;
}
注意事项:
1.以每个节点为起始统计总和。
页:
[1]