每日一题：Leetcode974.和可被k整除的子数组

MLGDOU2023-12-17 17:22

题目描述：

给定一个整数数组 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.在c++中，（负数%正数）=负数，所以需要对结果修正，（a%p+p）%p得到结果为正数。

2.同余定理：(a-b)%c==0<----->a%c==b%c

代码实现

cpp 复制代码

class Solution {
public:
    int subarraysDivByK(vector<int>& nums, int k) {
        int n=nums.size();
        unordered_map<int,int> mp;
        mp[0]=1;//0能整除k
        int sum=0;
        int res=0;
        for(auto x:nums)
        {
            sum+=x;
            int mod=(sum%k+k)%k;
            res+=mp[mod];
            mp[mod]++;
        }
        return res;
    }
};