LeetCode 437. Path Sum III 递归中使用递归,分两类情况

来源:互联网 发布:淘宝交易快照在哪里 编辑:程序博客网 时间:2024/06/05 08:24

437. Path Sum III

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值

注意

  • 和112. Path Sum有些类似,但是有些区别
  • 这个题的路径起始并不需要根节点到叶子结点,但是必须向下的

思路

在112.Path Sum中从上往下递归,是假定node在路径中,所以在这题中,应该分为两种情况。
1.当前的node在路径中
2.当前的node不在路径中

这里写图片描述

代码

class Solution {public:    // 在以root为根节点的二叉树中,寻找和为sum的路径,返回这样的路径个数    int pathSum(TreeNode* root, int sum) {        if(root == NULL)        {            return 0;        }        int res = 0;        //情况1:查找包含当前结点,是否存在其左右子树中路径其和为sum        res += findPath(root,sum);        //情况2:当前结点不包含在sum中,        res+=pathSum(root->left,sum);        res+=pathSum(root->right,sum);        return res;    }    private:    // 在以node为根节点的二叉树中,寻找包含node的路径,和为sum    // 返回这样的路径个数    int findPath(TreeNode* root,int num)    {        if(root == NULL)        {            return 0;        }        int res = 0;        if(root->val == num)  //当前结点的值等于num表示找到一条路径        {            res+=1;        }        res += findPath(root->left,num-root->val);        res += findPath(root->right,num-root->val);        return res;    }};

这里写图片描述

原创粉丝点击