leetcode 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的easy題，可是作起來仍是有點難度。。。

和以前從根開始找路徑的題比起來，須要多判斷以各個子結點爲起點計算路徑，遞歸實現。

 

代碼：

 1 /**
 2  * Definition for a binary tree node.
 3  * struct TreeNode {
 4  *     int val;
 5  *     TreeNode *left;
 6  *     TreeNode *right;
 7  *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 8  * };
 9  */
10 class Solution {
11 public:
12     int pathSumCore(TreeNode* root, int sum)
13     {
14         if(root == nullptr)
15             return 0;
16         int res=0;
17         if(root->val == sum)
18         {
19             res += 1;
20         }
21         res += pathSumCore(root->left, sum-root->val);
22         res += pathSumCore(root->right, sum-root->val);
23         return res;
24     }
25     int pathSum(TreeNode* root, int sum) {
26         if(root == nullptr)
27             return 0;
28         int res = 0;
29         res = pathSumCore(root, sum) + 
30             pathSum(root->left, sum) + pathSum(root->right, sum);
31         return res;
32     }
33 };