前缀和实例4(和可被k整除的子数组)

题目:

给定一个整数数组 nums 和一个整数 k ,返回其中元素之和可被 k 整除的(连续、非空) 子数组 的数目。

子数组 是数组的 连续 部分。

示例 1:

输入:nums = [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]

示例 2:

输入: nums = [5], k = 9
输出: 0

提示:

  • 1 <= nums.length <= 3 * 104
  • -104 <= nums[i] <= 104
  • 2 <= k <= 104

算法原理:

本题所需前置知识:

1 同余定理

如果 (a - b) % n == 0 ,那么可以得到⼀个结论: a % n == b % n 。即如果两个数相减的差能被n整除,那么这两个数对n取模的结果相同

如  (26 - 2) % 12 == 0 那么 26 % 12 == 2 % 12 == 2
 

2 c++ 中负数取模结果的修正:
 

c++ 中关于负数的取模运算,结果是「把负数当成正数,取模之后的结果加上⼀个负号」

如 -1 % 3 = -(1 % 3) = -1

由于余数在接下来的代码中会充当下标,所以余数不能为负数,故而有修正余数的情况:

(a % n + n) % n  

若是a%n的结果为负数,那么+n就会是正数,当然若是a%n的结果本身就是正数,那么+n就改变了,故而%n

当a%n的结果为负数,修正后依然比n小,故而%n不会发生改变

当a%n的结果为正数,无需修正,但+n使得结果变了,又%n,让变了的结果变回原样

 

前缀和实例4(和可被k整除的子数组)_第1张图片

枚举所有子数组可以每次固定一个起始位置向后枚举,我们当然也可以每次固定一个结尾位置向前枚举

i是任意位置,那么以 i 为结尾的和可被k 整除的子数组个数就可以这么求:

由图示,我们知道要求[0,i]区间内绿线的个数,只要知道红线的个数即前缀和为x的个数就可以了,又sum%k==x%k

那么我们只需要知道在[0,i-1]内,前缀和(即x)%k==sum%k的个数即可

hash表统计前缀和%k的余数出现的次数

细节问题:hash[0] = 1 ,当i位置的前缀和本身sum就是能被k整除的,[0,i]区间本身就是一个合法子数组,那么x=0

代码实现:

class Solution 
{
public:
    int subarraysDivByK(vector& nums, int k) 
    {
        unordered_map hash;
        hash[0] = 1;//0这个数(这个前缀和)的余数,也可以写成hash[0%k]=1
        int ret = 0;
        int sum = 0;
        for(auto e:nums)
        {
            sum+=e;//当前位置的前缀和
            int r = (sum%k+k)%k;//修正后的余数
            if(hash.count(r))
            {
                ret+=hash[r];
            }
            hash[r]++;
        }
        return ret;
    }
};

你可能感兴趣的:(算法合集,算法,leetcode)