給定一個二叉樹,它的每個結點都存放着一個整數值。
找出路徑和等於給定數值的路徑總數。
路徑不需要從根節點開始,也不需要在葉子節點結束,但是路徑方向必須是向下的(只能從父節點到子節點)。
二叉樹不超過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
思路:"路徑不需要從根節點開始,也不需要在葉子節點結束,但是路徑方向必須是向下的",說明必須要遍歷每個結點,然後有些特殊情況,比如題目中“5->3”,沒想到什麼好辦法,後來參考別人的想法,只需要保存每個根節點到當前結點的路徑,然後依次從上到下的踢除一些結點就可以了。
void helper(TreeNode* root,int sum,int cursum,vector<TreeNode*>& path_vec,int& count){
if(!root) return;
cursum += root->val;
path_vec.push_back(root);
if(cursum == sum) count++;
int tempsum = cursum;
for(int i = 0;i<path_vec.size()-1;i++){
tempsum -= path_vec[i]->val;
if(sum == tempsum) count++;
}
helper(root->left,sum,cursum,path_vec,count);
helper(root->right,sum,cursum,path_vec,count);
path_vec.pop_back(); // -- 1
}
class Solution {
public:
int pathSum(TreeNode* root, int sum) {
int count =0,cursum = 0;
vector<TreeNode*> path;
helper(root,sum,cursum,path,count);
return count;
}
};
至於1步驟的講解,別人給出的詳細解答:437.路徑總和III
總結:對樹的遞歸還是要多做題多思考,特別是遞歸的整個過程要通過做題來思考,以及引發的cursum和path_vec,什麼時候用引用,什麼時候不用,有了更深的理解。