leetcode560.和为k的子数组

leetcode560.和为k的子数组_第1张图片

前缀和+哈希表

 leetcode560.和为k的子数组_第2张图片

class Solution {
public:
    int subarraySum(vector& nums, int k) {
        int result=0;
        int pre=0;
        unordered_map map;
        map[0]=1;//模拟测试用例1[1,1,1],k=2可知
        for(const int& num:nums){
            pre+=num;
            if(map.find(pre-k)!=map.end())
                result+=map[pre-k];
            map[pre]++;
        }
        return result;
    }
};

 

你可能感兴趣的:(前缀和,哈希表,leetcode)