leetcode前缀和哈希表(974. 和可被 K 整除的子数组)

盘门

已于 2024-08-26 23:05:51 修改

阅读量216

点赞数 1

文章标签：算法 java 数据结构

于 2024-08-26 23:01:06 首次发布

本文链接：https://blog.csdn.net/acuteeagle01/article/details/141575595

版权

前言

经过前期的基础训练以及部分实战练习，粗略掌握了各种题型的解题思路。现阶段开始专项练习。

描述

给定一个整数数组 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

实现原理与步骤

前缀和：考虑数组中从头到第 i 个元素的和。如果两个前缀和的差能被 K 整除，那么这两个前缀和之间的子数组的和也能被 K 整除。
同余关系：我们可以通过前缀和对 K 取模的结果来确定是否存在符合条件的子数组。如果两个前缀和 prefixSum1 和 prefixSum2 对 K 取模的结果相同，即 (prefixSum1 % K) == (prefixSum2 % K)，那么 prefixSum1 和 prefixSum2 之间的子数组的和就是 K 的倍数。
哈希表记录：用一个哈希表来记录前缀和对 K 取模的结果及其出现的次数。每当遇到一个新的前缀和时，检查哈希表中是否存在相同取模结果的前缀和，如果有，则说明存在满足条件的子数组。

实现代码

class Solution {
    public int subarraysDivByK(int[] nums, int k) {
        int sum=0;
        int res=0;
        HashMap<Integer,Integer> map=new HashMap();
        // 初始化：前缀和为0时，其模K的结果为0，次数为1
        //由于第一次碰到mod为0，则res要加1.mod不为0情况则只需在map中标记为0.
        map.put(0, 1);
        for(int i=0;i<nums.length;i++){
            sum+=nums[i];
            //对负数进行处理
            int mod=(sum%k+k)%k;
            if(map.containsKey(mod)){
                //加上mod相等的个数，当前mod为1，则之前的mod为1的位置与当前位置之间的间隔都为k
                res+=map.get(mod);
            }
            //更新mod的数量
            map.put(mod,map.getOrDefault(mod,0)+1);
        }
        return res;
    }
}