糖逗 发表于 2020-4-24 13:09:38

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]
查看完整版本: C++刷leetcode(面试题 04.12. 求和路径)【深度优先搜索】