leetcode热题精选:437. 路径总和 III

437. 路径总和 III

给定一个二叉树，它的每个结点都存放着一个整数值。

找出路径和等于给定数值的路径总数。

路径不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。

二叉树不超过1000个节点，且节点数值范围是 [-1000000,1000000] 的整数。

示例：

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

  10
 /  \
5   -3

/ \
3 2 11
/ \
3 -2 1

返回 3。和等于 8 的路径有:

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/path-sum-iii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

代码

下面是同一种方法，用了两种函数的接口来实现，一种是带着一个数遍历，另外一个是通过计数有返回值。

    int pathSum(TreeNode* root, int sum) {
        if(root==NULL)return 0;
        int count=0;
        helper(root,sum,count);
        int lres,rres;
        lres=pathSum(root->left,sum);
        rres=pathSum(root->right,sum);
        return count+lres+rres;
    }
    void helper(TreeNode* T,int sum,int& count){
        if(T==NULL)return;
        int res=sum-T->val;
        if(res==0)count++;
        helper(T->left,res,count);
        helper(T->right,res,count);
    }

  int helper2(TreeNode* T,int sum){
      if(T==NULL)return 0;
      int count,lsum,rsum;
      count=0;
      if(T->val==sum)count++;
      lsum=helper2(T->left,sum-T->val);
      rsum=helper2(T->right,sum-T->val);
      count=count+lsum+rsum;
      return count;
  }
  
  int pathSum(TreeNode* root, int sum) {
      if(root==NULL)return 0;
      int count=helper2(root,sum);
      int lres,rres;
      lres=pathSum(root->left,sum);
      rres=pathSum(root->right,sum);
      return count+lres+rres;
  }

leetcode112. 路径总和

之前我写的其他解法：

 /*递归的解法*/
 bool hasPathSum(TreeNode* root, int sum) {
        if(root==NULL) return false;
        if(root->left==NULL&&root->right==NULL) 
            return sum==root->val;
        else
            return (hasPathSum(root->left,sum-root->val)||hasPathSum(root->right,sum-root->val));
    }        
    ```
    ```
		/*这个题目也可以用来训练如何使用栈来遍历树*/
		class Solution {
public:
    bool hasPathSum(TreeNode* root, int sum) {
        stack<pair<TreeNode*,int>> s;
        TreeNode* temp=root;
        while(!s.empty()||temp!=NULL){
            while(temp!=NULL){
                s.push(make_pair(temp,sum-temp->val));
                sum=sum-temp->val;
                if(sum==0&&temp->left==NULL&&temp->right==NULL) return true;//除了判断和，还需要判断是否是叶节点，用来针对【1 2】，1的这种情况
                temp=temp->left;
            }
            auto item=s.top();
            s.pop();
            temp=item.first;
            sum=item.second;
            temp=temp->right;
        }
    return false;
    }
};

栈解法的修改

class Solution {
public:
    bool hasPathSum(TreeNode* root, int sum) {
        if(!root)return false;
        stack<pair<TreeNode*,int>> s;
        s.push(make_pair(root,sum));
        TreeNode* T;
        int num;
        while(!s.empty()){
            auto iter=s.top();
            s.pop();
            T=iter.first;
            num=iter.second;
            if(T->left==NULL&&T->right==NULL){
                if(T->val==num)return true;
            }
            else{
                if(T->left!=NULL)s.push(make_pair(T->left,num-T->val));
                if(T->right!=NULL)s.push(make_pair(T->right,num-T->val));
            }
        }
        return false;
    }
};