LeetCode笔记---同余定理

原创已于 2023-10-03 20:49:13 修改 · 434 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#leetcode

于 2022-06-08 13:46:13 首次发布

LeetCode 专栏收录该内容

5 篇文章

订阅专栏

该篇博客讨论了一种算法问题，即找出数组中所有元素之和能被整数K整除的连续子数组的数量。由于数据量较大，直接使用前缀和会导致时间复杂度过高，因此提出了利用前缀和与哈希表的方法。通过计算每个前缀和对K取模后的值，并统计这些模值的出现次数，可以高效地计算出满足条件的子数组组合。博客提供了具体的Java代码实现，并解释了算法思路。

更多看这里吧😁

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

给定一个整数数组 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 * 10^4
-104 <= nums[i] <= 104 2 <= k <= 10^4

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/subarray-sums-divisible-by-k
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

数据量为1 <= nums.length <= 3 * 10^4，直接使用前缀和+差分
时间复杂度O(n^2),会超时。

当i <= j，S[i] - S[ j - 1] 表示前缀和 a[ j ] + a[ j + 1 ] + … a[ i ]
如果前缀和是K的倍数：
则 ( S[i] - S[ j - 1] ) mod k == 0
根据同余定理：
得到 ( S[i] ) mod k - ( S[ j - 1] ) mod k == 0
得到 ( S[i] ) mod k == ( S[ j - 1] ) mod k
所以只要在前缀和数组中统计 t = ( S[i] ) mod k 的个数，相同t可以两两组合，利用组合公式算出该t的所有组合，再将不同t的数量相加即可。

在这里插入图片描述
其中m = 2, n等于每个t内的数量

C n m = n ! / (2 ! * (n - 2) ! ) = n * (n - 1) / 2

class Solution {
    public int subarraysDivByK(int[] nums, int k) {
        int[] map = new int[k];
        int p = 0;
        int[] prefixSum = new int[nums.length + 1];
        for (int i = 1; i < prefixSum.length; i++) {
            prefixSum[i] = p + nums[i - 1];
            p = prefixSum[i];
        }
        for (int i = 0; i < prefixSum.length; i++) {
            map[ prefixSum[i] % k < 0 ? (prefixSum[i] % k + k) : 
            prefixSum[i] % k ]++;
        }
        int ans = 0;
        for (int i = 0; i < map.length; i++) {
            ans += map[i] * (map[i] - 1) / 2;
        }
        return ans;
    }
}

更多看这里吧😁