和可被 K 整除的子数组（LeedCode）

最新推荐文章于 2024-09-13 16:27:15 发布

好好学习Py

最新推荐文章于 2024-09-13 16:27:15 发布

阅读量325

点赞数 7

分类专栏：算法与数据结构文章标签：算法数据结构 leetcode

本文链接：https://blog.csdn.net/weixin_74254879/article/details/141231493

版权

算法与数据结构专栏收录该内容

82 篇文章 0 订阅

订阅专栏

题目

给定一个整数数组 nums 和一个整数 k ，返回其中元素之和可被 k 整除的非空 子数组 的数目。

子数组 是数组中连续的部分。

解题

def subarraysDivByK(nums, k):
    count_map = {0: 1}  # 余数为0的初始情况
    current_sum = 0
    result = 0

    for num in nums:
        current_sum += num
        mod = current_sum % k

        # 处理负数的情况
        if mod < 0:
            mod += k

        # 统计当前余数mod出现的次数
        if mod in count_map:
            result += count_map[mod]
            count_map[mod] += 1
        else:
            count_map[mod] = 1

    return result


nums = [4, 5, 0, -2, -3, 1]
k = 5
print(subarraysDivByK(nums, k))  # 输出: 7