Leetcode | 437. 路径总和 III

Leetcode | 437. 路径总和 III

给定一个二叉树，它的每个结点都存放着一个整数值。
找出路径和等于给定数值的路径总数。
路径不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。
二叉树不超过1000个节点，且节点数值范围是 [-1000000,1000000] 的整数。
示例

题目解释

由题意可知，即从二叉树找随意找到一个由相连节点组成的路径的和等于sum值即可，找出所有可能的路径个数，返回总和。

解答

• 这种类型的题目，个人反应肯定第一解决方案选择深度优先搜索。
• 当当前节点的值刚好等于sum值，则返回1个可能值，如果不等，则更新新的sum值，即sum=sum-node->val
• 但是也要注意一点的就是，题目中说到不需要从根节点开始，也不需要在叶子节点结束，所以二叉树中的所有非空节点都可称为深度优先搜索的起始节点。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int pathSum(TreeNode* root, int sum) {
        // 如果当前节点为空，则相当于未找到路径，返回0
        if (!root)
            return 0;

        // 因为题目说，不要求一定从根节点开始和不一定从叶子节点结束
        // 所以，二叉树中的所有节点都可以当作dfs的其实节点。
        return dfs(root, sum) + pathSum(root->left, sum) + pathSum(root->right, sum);
    }

    int dfs(TreeNode* root, int sum){
        // 如果当前节点为空，则相当于未找到路径，返回0
        if (!root)
            return 0;

        // 如果当前节点的值刚好等于目标值，则返回加1
        if (root->val == sum)
            return 1 + dfs(root->left, 0) + dfs(root->right, 0);

        // 如果当前节点的值不等于目标值，则更新新的目标值new_sum = sum - root->val;
        return dfs(root->left, sum - root->val) + dfs(root->right, sum - root->val);
    }
};