LeetCode 437 Path Sum III 题解

来源:互联网 发布:手机识图软件 编辑:程序博客网 时间:2024/06/06 08:29

题意简述:给定一棵二叉树,以及一个和sum,问在树中是否存在一条路径,路径上所有结点之和恰好等于sum。路径不限制一定要开始于根结点或者结束于叶子结点,但是一定要是向下的(从父亲结点向儿子结点)。如果存在这样的路径,求出有多少条。
输入:二叉树的根结点指针root,以及和sum
输出:满足条件的路径数
示例:对于下图的二叉树,sum为8,则存在3条路径:
    10
   /  \        1. 5 -> 3
   5   -3
   /  \    \       2. 5 -> 2 -> 1
  3   2   11
 /  \    \         3. -3 -> 11
3   -2   1


题解:显而易见的递归问题,递归的对象是路径的数目。
对于一个根结点,选择它是否做为路径起点:

  • 如果不选择它作为路径起点,则当前的sum不变,其左右子树继续进行这个递归;
  • 如果选择它作为路径起点,则对于它以及之后递归的过程,sum减去当前结点的值,左右子树继续递归;(这是因为已经选择了路径起点的话,后来的所有结点都要选择,否则这条路径就不连续,与题意不符)
  • 递归终止于叶子结点。

存在路径的条件是在当前递归中sum的值与结点的值相等,这样路径数+1,但是这不是递归的终止条件。因为结点的值存在负数,有可能在这个已发现的路径上再增加几个结点又是一条路径,所以发现一条路径应该继续递归,直至遇到叶子结点。

实现的细节如下:

/** * 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 {private:    int noChoosePath(TreeNode* root, int sum) {        if(root == NULL) return 0;        return noChoosePath(root->left, sum) + noChoosePath(root->right, sum) + choosePath(root, sum);    }    int choosePath(TreeNode* root, int sum) {        if(root == NULL) return 0;        if(root -> val == sum)            return 1 + choosePath(root->left, sum-root->val)             + choosePath(root->right, sum-root->val);        else            return choosePath(root->left, sum-root->val)             + choosePath(root->right, sum-root->val);    }public:    int pathSum(TreeNode* root, int sum) {        return(noChoosePath(root, sum));    }};
0 0
原创粉丝点击