1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32
|
class Solution { public: int func(TreeNode* root, int targetSum, long long cur, unordered_map<long long, int>& map) { if (!root) return 0; cur = cur + root->val; int count = map[cur - targetSum]; map[cur]++; count = count + func(root->left, targetSum, cur, map); count = count + func(root->right, targetSum, cur, map); map[cur]--; return count; } int pathSum(TreeNode* root, int targetSum) { unordered_map<long long, int> map; map[0] = 1; return func(root, targetSum, 0, map); } };
|