天天看點

JZ84 二叉樹中和為某一值的路徑(三)

/**
 * struct TreeNode {
 * int val;
 * struct TreeNode *left;
 * struct TreeNode *right;
 * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
class Solution {
public:
    /**
     * 代碼中的類名、方法名、參數名已經指定,請勿修改,直接傳回方法規定的值即可
     *
     * 
     * @param root TreeNode類 
     * @param sum int整型 
     * @return int整型
     */
   int ans = 0;
    void dfs (TreeNode*r, int sum) {
        if (!r) return;
        sum -= r->val;
        if (sum == 0) {
            ans ++;
        }
        dfs(r->left, sum);
        dfs(r->right, sum);
    }
    int FindPath(TreeNode* root, int sum) {
        if (!root) return ans;
        dfs(root, sum);
        FindPath(root->left, sum);
        FindPath(root->right, sum);
        return ans;
    }
};