LeetCode每日一题5月27日 LeetCode974. 可被K整除的子数组

问题描述：

给定一个整数数组 A，返回其中元素之和可被 K 整除的（连续、非空）子数组的数目。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/subarray-sums-divisible-by-k

示例：

输入：A = [4,5,0,-2,-3,1], K = 5
输出：7
解释：
有 7 个子数组满足其元素之和可被 K = 5 整除：
[4, 5, 0, -2, -3, 1], [5], [5, 0], [5, 0, -2, -3], [0], [0, -2, -3], [-2, -3]


提示：

1 <= A.length <= 30000
-10000 <= A[i] <= 10000
2 <= K <= 10000

解题思路：

暴力解法，双层循环，

第一层循环计算 A[0]-A[i-1]的值，i<=A.size();

    第二层循环计算A[0]-A[j-1]的值，j

完成代码如下：

class Solution {
public:
    int subarraysDivByK(vector& A, int K) {
        int res = 0;
        int sum = 0;
        for(int i=0;i

但是这个解法有个测试用例超出了运行时间

所以我们需要进行一些优化，优化思路借助评论区大神和LeetCode560，和为k的子数组

运用前缀和的思想。

我们首先要知道一个原理，就是在原数组上面任何位置任何数加上n*K(n是整数)，对结果不会产生影响

即和为元素 a 和 a+n*k俩种情况是等价的。

 

class Solution {
public:
    int subarraysDivByK(vector& A, int K) {
        int res = 0;
        int sum = 0;
        map m;
        m[0]=1;
        for(int i=0;i