題目大意:找出二叉樹中路徑和等於目標和的路徑個數,節點爲整數(可能爲負),路徑起點不一定爲根節點,終點不一定爲葉節點,但是路徑方向始終向下
分析:二叉樹dfs。相比leetcode112、113要注意三點:節點數值可能爲負,路徑起點可以任選,路徑終點不一定是葉節點。
因爲路徑起點任選,所以要以樹中每一個節點爲起點dfs,也就是用countHelper統計以該點爲起點的路徑條數,然後遞歸統計左右節點爲起點的路徑條數。
countHelper統計路徑條數:遞歸過程中記錄當前路徑和,如果路徑和等於目標值,那麼路徑數加一,注意由於節點值可能爲負,所以此時不能return,還要繼續遞歸下去。
由於路徑終點不一定是葉節點,所以符合目標的條件比leetcode112、113少了一個,不用判斷節點左右孩子是否爲空了。
代碼:
/**
* 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 {
int ans = 0;
public:
int pathSum(TreeNode* root, int sum) {
if(!root) return 0;
countHelper(root,0,sum);
pathSum(root->left,sum);
pathSum(root->right,sum);
return ans;
}
void countHelper(TreeNode* root, int cur, int sum){
if(!root) return;
cur += root->val;
if(cur == sum){
ans++;
}
countHelper(root->left,cur,sum);
countHelper(root->right,cur,sum);
}
};