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

 

思路:"路徑不需要從根節點開始,也不需要在葉子節點結束,但是路徑方向必須是向下的",說明必須要遍歷每個結點,然後有些特殊情況,比如題目中“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,什麼時候用引用,什麼時候不用,有了更深的理解。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章