【Leetcode】974. Subarray Sums Divisible by K

题目地址：

https://leetcode.com/problems/subarray-sums-divisible-by-k/

给定一个长$n$的数组$A$，再给定一个大于$1$的正整数$k$，问其有多少个非空子数组和是$k$的倍数。

求一下$A$的前缀和模$k$数组$c$，即$c[i]=({\sum }_{k=0}^{i-1}A[k])\mathrm{mod}k$，接着遍历$c$，并开一个哈希表，key是模$k$的值，value是模$k$的值是key的前缀有多少个。这样遍历到$c[i]$的时候，查一下哈希表key为$c[i]$的value，就知道了以$A[i-1]$为右端点的和是$k$的倍数的子数组有多少个了，累加即可。代码如下：

class Solution {
 public:
  int subarraysDivByK(vector<int>& a, int k) {
    int res = 0;
    unordered_map<int, int> mp;
    mp[0] = 1;
    for (int i = 0, s = 0; i < a.size(); i++) {
      s = (s + a[i] % k + k) % k;
      res += mp[s];
      mp[s]++;
    }
    return res;
  }
};

时间复杂度$O(n)$，空间$O(k)$。