leetcode437——Path SumⅢ

題目大意:找出二叉樹中路徑和等於目標和的路徑個數,節點爲整數(可能爲負),路徑起點不一定爲根節點,終點不一定爲葉節點,但是路徑方向始終向下

分析:二叉樹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);
    }
};

 

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