LeetCode刷题笔记（Path Sum III）

刚刚又刷了一道题，感觉难度还是有一些的，不知道为啥标注的是easy，难道是笔者的水平在下降吗？下面就和大家分享一下经验吧！

题目如下：

You are given a binary tree in which each node contains an integer value.

Find the number of paths that sum to a given value.

The path does not need to start or end at the root or a leaf, but it must go downwards (traveling only from parent nodes to child nodes).

The tree has no more than 1,000 nodes and the values are in the range -1,000,000 to 1,000,000.

Example:

root = [10,5,-3,3,2,null,11,3,-2,null,1], sum = 8

      10
     /  \
    5   -3
   / \    \
  3   2   11
 / \   \
3  -2   1

Return 3. The paths that sum to 8 are:

1.  5 -> 3
2.  5 -> 2 -> 1
3. -3 -> 11

题意分析： 

给定一个每个节点值都是整数(可正可负)的二叉树，请计算满足节点值之和为sum的路径个数，其中路径不一定要一定从根节点到叶子节点，但路径必须是向下走的，即从父节点到子节点。

方法一（递归法）

本题的难点在于：①节点值可正可负，导致满足节点值之和为sum的路径，其子路径也有可能满足节点值之和为sum；②路径的起点与终点不一定是根节点和叶子节点。

本题需要分类讨论，第一种情况：路径包含根节点，第二种情况：路径不包含根节点。对于第一种情况，新建函数findpath用于找到满足条件的路径，直接将根节点root与sum值传给findpath进行递归调用。对于第二种情况，分别将根节点的左右子节点与sum值传给findpath进行递归调用。最后将两个情况递归调用的结果相加并返回即可。

解题代码如下：

class Solution{
public:
    int pathSum(TreeNode* root, int sum){
        if (root==NULL) return 0;
        int res = findpath(root, sum);
        res += pathSum(root->right, sum);
        res += pathSum(root->left, sum);
        return res;
    }

    int findpath(TreeNode* root, int sum){
        if(root==NULL) return 0;
        int res = 0;
        if(root->val - sum == 0) res++;
        res += findpath(root->right, sum - root->val);
        res += findpath(root->left, sum - root->val);
        return res;
    }
};

提交后的结果如下：

 

方法二（递归法）

采用两个递归函数，一个是sumFindpath函数，以当前结点起点，采用前序遍历，对于每个遍历到的节点进行处理，用pre_value来记录当前路径之和，用cur_value来记录pre_value加上当前节点值，若cur_value等于sum则返回结果时要加1，然后再对当前节点的左右子节点进行递归调用求解。另一个是pathSum函数，对当前结点调用sumFindpath函数，加上对左右子结点调用pathSum递归函数，最后将三部分的结果相加即为所求结果。

解题代码如下：

class Solution {
public:
    int pathSum(TreeNode* root, int sum) {
        if (!root) return 0;
        return sumFindpath(root, 0, sum) + pathSum(root->left, sum) + pathSum(root->right, sum);
    }
    int sumFindpath(TreeNode* node, int pre_value, int& sum) {
        if (!node) return 0;
        int cur_value = pre_value + node->val;
        return (cur_value == sum) + sumFindpath(node->left, cur_value, sum) + sumFindpath(node->right, cur_value, sum);
    }
};

提交后的结果如下：

 

 

 

日积月累，与君共进，增增小结，未完待续。