leetcode【每日一题】974. 和可被 K 整除的子数组 Java

最新推荐文章于 2022-05-02 15:10:54 发布

宰了那只汤姆猫

最新推荐文章于 2022-05-02 15:10:54 发布

阅读量239

点赞数

分类专栏： leetcode刷题

本文链接：https://blog.csdn.net/qq_43491066/article/details/106375503

版权

leetcode刷题专栏收录该内容

476 篇文章 3 订阅

订阅专栏

题干

给定一个整数数组 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 <= A.length <= 30000
-10000 <= A[i] <= 10000
2 <= K <= 10000

想法

第一想法暴力
把所有枚举出来再判断，显然不行

然题干上说了是连续的，那么可以想到前缀法
设P(i)表示0到i位置的和
那么(i,j)这个区间的判断子数组的和能否被 KK 整除就等价于判断 (P[j] - P[i-1]) \bmod K == 0
所以就维护所有的p(i)，即P[j]modKP[i−1]modK
再判断哪些P[j]modKP[i−1]modK即可
那么我使用一个hahsmap
key是P[j]modK，value是P[j]modK在这个数组里出现了几次。

然后统计有多少个一样的就可以了

Java代码

package daily;

import java.util.HashMap;

public class SubarraysDivByK {
    public int subarraysDivByK(int[] A, int K) {
        //子列和  ，结果，模，相同个数
        int sum=0,count=0,mod,same;
        HashMap<Integer,Integer> map=new HashMap<>();
        //初始化
        map.put(0,1);
        for (int num:A
             ) {
            //子列和
            sum+=num;
            //负数需要纠正
            mod=(sum % K + K) % K;
            //模相同的个数
            same=map.getOrDefault(mod,0);
            count+=same;
            map.put(mod,same+1);



        }
        return  count;

    }


    public static  void main(String[] args){

        int []A = {4,5,0,-2,-3,1};
        int K=5;
        SubarraysDivByK subarraysDivByK=new SubarraysDivByK();
        System.out.println(subarraysDivByK.subarraysDivByK(A,K));
    }

}