2020-05-27 LeetCode 974 和可被 K 整除的子数组 C

题目:和可被 K 整除的子数组

 给定一个整数数组 A，返回其中元素之和可被 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. 1 <= A.length <= 30000
2. -10000 <= A[i] <= 10000
3. 2 <= K <= 10000

 今天也是郁闷，本来题目挺简单的，自己瞎改测试用例用来debug废了好多时，一直报内存泄漏的错误，最后发现是控制台K给该没了，大意。

 题目和560很像，思路是前缀和+同余定理，用长度为K的数组记录即可，需要注意的是前缀和的余数是负数的时候需要+K，遍历过程中如果和的余数为0直接结果直接+1，其他则加同余数的前缀和个数，第一版如下

int subarraysDivByK(int* A, int ASize, int K){

    int* const cnt=(int*)malloc(sizeof(int)*K);
    memset(cnt, 0, sizeof(int)*K);
    int i,ans=0,sum=0;
    for(i=0;i<ASize;i++){
        sum += A[i];
        sum = (sum % K + K) % K;
        if(sum==0)
            ans++;
        ans+=cnt[sum];
        cnt[sum]++;
    }
    free(cnt);
    return ans;
}

运行结果:

 后面进行了改进，把初始时候的cnt[0]置1就没必要每次判断和是否为0了，改进后如下

int subarraysDivByK(int* A, int ASize, int K){
    int* const cnt = (int*)malloc(sizeof(int) * K);
    memset(cnt, 0, sizeof(int) * K);
    int i,ans=0,sum=0;
    cnt[0]=1;
    for(i=0;i<ASize;i++){
        sum += A[i];
        sum = (sum % K + K) % K;
        ans+=cnt[sum];
        cnt[sum]++;
    }
    free(cnt);
    return ans;
}

运行结果:

 个人感觉官方题解用HashMap的方式还可以优化，像560一样如果能用数组表示的情况下用HashMap，虽然时间复杂度都是O(N)，时间耗费肯定是会增加的。