LeetCode974. 和可被 K 整除的子数组

最新推荐文章于 2020-05-27 21:04:43 发布

香榭的落叶l

最新推荐文章于 2020-05-27 21:04:43 发布

阅读量149

点赞数

分类专栏： LC&Coder

本文链接：https://blog.csdn.net/qq_42007742/article/details/106377823

版权

LC&Coder 专栏收录该内容

61 篇文章 0 订阅

订阅专栏

该博客主要介绍LeetCode第974题的解决方案。题目要求找出整数数组中元素之和能被K整除的连续、非空子数组的个数。博主指出这是一道与560题相似的题目，可以通过前缀和结合同余定理来解决。在实现中，博主利用一个哈希表存储前缀和模K的结果及其出现次数，当遇到相同模K的结果时，就找到了一个满足条件的子数组。

摘要由CSDN通过智能技术生成

题目

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

思路

560题的换皮题目，想到了前缀和不知道同余定理。。简单说就是

$m$ 是大于1的整数， $a$ 、 $b$ 是整数，如果 $m ∣ (a - b)$ ，则称 $a$ 与 $b$ 关于模 $m$ 同余。
由定义，(a-b) mod m == 0 -> a mod == b mod m

用map存储每个前缀和，以模结果为key，出现的次数为value，遇到相同的key即找到了一个满足和被K整除的子串。

class Solution {
    public int subarraysDivByK(int[] A, int K) {
        Map<Integer,Integer>map=new HashMap<>();
        int sum=0,res=0;
        //put(0,1)表示当前前缀和本身被K整除
        map.put(0,1);
        for(int i=0;i<A.length;i++){
            sum+=A[i];
            //java的%运算时取余
            int mod=Math.floorMod(sum,K);
            int sameMod=map.getOrDefault(mod,0);
            res+=sameMod;
            map.put(module,sameMod+1);
        }
        return res;
    }
}