力扣974被K整除的子数组

同余定理

力扣974被K整除的子数组_第1张图片

使用前缀和+哈希表

力扣974被K整除的子数组_第2张图片
力扣974被K整除的子数组_第3张图片
由于可能是负数所以要进行修正:(sum%k+k)%k

class Solution {
public:
    int subarraysDivByK(vector<int>& nums, int k) {
        unordered_map<int,int> hash;
        hash[0 % k] = 1;   //0 这个数的余数

        int sum = 0, ret = 0;
        for(auto x : nums)
        {
            sum += x;
            int r = (sum % k + k) % k;
            if(hash.count(r)) ret += hash[r];
            hash[r]++;
        }
        return ret;
    }
};

你可能感兴趣的:(刷题,leetcode,哈希算法,散列表)