Leetcode.974 和可被 K 整除的子数组

题目链接

Leetcode.974 和可被 K 整除的子数组 rating : 1676

题目描述

给定一个整数数组 n u m s nums nums 和一个整数 k k k ,返回其中元素之和可被 k 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 ≤ n u m s . l e n g t h ≤ 3 ∗ 1 0 4 1 \leq nums.length \leq 3 * 10^4 1nums.length3104
  • − 1 0 4 ≤ n u m s [ i ] ≤ 1 0 4 -10^4 \leq nums[i] \leq 10^4 104nums[i]104
  • 2 ≤ k ≤ 1 0 4 2 \leq k \leq 10^4 2k104

解法:前缀和 + 哈希表

我们假设 [ j , i ] [j,i] [j,i] 区间的子数组元素和可以被 k k k 整除,即 :

( n u m s [ j ] + n u m s [ j + 1 ] + . . . + n u m s [ i − 1 ] + n u m s [ i ] )   m o d   k = 0 (nums[j] + nums[j + 1] + ... + nums[i-1] + nums[i])\ mod\ k = 0 (nums[j]+nums[j+1]+...+nums[i1]+nums[i]) mod k=0

我们用 s u m sum sum 表示 n u m s nums nums 的前缀和数组,可将上式转换为:

( s u m [ i ] − s u m [ j − 1 ] )   m o d   k = 0 (sum[i] - sum[j-1])\ mod\ k = 0 (sum[i]sum[j1]) mod k=0

再转换一下得到:

s u m [ j − 1 ]   m o d   k = s u m [ i ]   m o d   k sum[j-1]\ mod\ k= sum[i]\ mod\ k sum[j1] mod k=sum[i] mod k

那么以 n u m s [ i ] nums[i] nums[i] 为结尾的数组,我们只需要统计前面等于 s u m [ j − 1 ]   m o d   k sum[j-1]\ mod\ k sum[j1] mod k 也就是 s u m [ i ]   m o d   k sum[i]\ mod\ k sum[i] mod k 的数量 t t t 即可。

那么这个 t t t 就是以 n u m s [ i ] nums[i] nums[i] 为结尾的数组中 和能被 k k k 整除的子数组的数量。

我们只需要对每一个 n u m s [ i ] nums[i] nums[i] 都加上 t t t 即可,这样我们就可以统计出所有的 和能被 k k k 整除的子数组的数量。

在实现上,我们使用哈希表来记录前缀和出现的次数。初始时,和为 0 0 0 ,也需要统计它的出现次数,即 { 0 , 1 } \{ 0 , 1 \} {0,1}

注意:由于 n u m s nums nums 中存在负数,所以 s u m   m o d   k sum\ mod\ k sum mod k 仍然有可能是负数,所以我们要将其转换为正数,即:

k e y = ( s u m   m o d   k + k )   m o d   k key = (sum\ mod\ k + k)\ mod\ k key=(sum mod k+k) mod k

时间复杂度: O ( n ) O(n) O(n)

C++代码:

class Solution {
public:
    int subarraysDivByK(vector<int>& nums, int k) {
        int n = nums.size();
        int ans = 0 , sum = 0;
        unordered_map<int,int> cnt{{0,1}};

        for(int i = 0;i < n;i++){
            sum += nums[i];
            auto key = (sum % k + k) % k;
            ans += cnt[key];
            cnt[key]++;
        }

        return ans;
    }
};
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值