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

关关的刷题日记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);
    }
};

照顾好自己的身体,控制好自己的情绪,加油!

展开全文
相关主题
Top
微信扫码咨询专知VIP会员