LeetCode 560. Subarray Sum Equals K

原题目：https://leetcode-cn.com/problems/subarray-sum-equals-k/

 

思路：

pre【i】代表了从0到i的元素和。

则【j。。。。i】的和 为pre【i】-pre【j-1】，建立哈希表，以从0到i的元素和pre【i】为键值，出现的次数为对应的值。

我们只需要找到map【pre【i】-k】的可数就行了。

注意边界条件：取map【0】=1；（可以考虑[3,4],k=7的情况）。

 

代码：

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        map<int, int> mp;
        mp[0] = 1;
        int count = 0, pre = 0;
        for (auto& x:nums) {
            pre += x;
            if (mp.find(pre - k) != mp.end()) count += mp[pre - k];
            mp[pre]++;
        }
        return count;
    }
};