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

运行结果:
2020-05-27 LeetCode 974 和可被 K 整除的子数组 C_第1张图片

 后面进行了改进,把初始时候的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

运行结果:

2020-05-27 LeetCode 974 和可被 K 整除的子数组 C_第2张图片
 个人感觉官方题解用HashMap的方式还可以优化,像560一样如果能用数组表示的情况下用HashMap,虽然时间复杂度都是O(N),时间耗费肯定是会增加的。

你可能感兴趣的:(C,LeetCode,leetcode,算法)