【关关的刷题日记60】Leetcode 437. Path Sum III- 专知

会员服务 ·

【关关的刷题日记60】Leetcode 437. Path Sum III

关关

点击上方“专知”，关注获取专业AI知识”

关关的刷题日记60 – Leetcode 437. Path Sum III

题目

题目的意思是给定一个二叉树，让我们找到路径节点和等于给定值的路径的个数。这里的路径不一定是从根节点到叶子节点，可以是其中的一段，但是必须是自上而下的顺序。

思路

思路：这题目不好做，涉及到递归的嵌套。一棵树中满足路径节点和等于给定值的路径的个数，等于以这棵树的根节点为路径源头的满足要求的个数，加上它的左右子树中满足要求的树的个数。左右子树中满足要求的树的个数的求法，和这棵树中满足要求的个数的求法一样，这是第一层递归。以这棵树的根节点为路径源头的满足要求的个数的做法，同关关的刷题日记58 – Leetcode 112 Path Sum中的做法一样，只不过不是到叶子节点，才判断节点值是否与sum值相等，而是路径中的任一点都可以。

class Solution {public: 
    int dfs(TreeNode* root, int sum)
    {
        int count=0;
        if(root==nullptr)
            return 0;
        if(root->val==sum)
            count++;
        count+=dfs(root->left,sum-root->val);
        count+=dfs(root->right,sum-root->val);
        return count;
    }

    int pathSum(TreeNode* root, int sum) {
        if(root==nullptr)
            return 0;
        return dfs(root,sum)+pathSum(root->left,sum)+pathSum(root->right,sum);
    }};