和可被 K 整除的子数组

/*
 * 思路：
 * 1. 使用前缀和数组来计算连续子数组的和。
 * 2. 利用 HashMap 来存储当前前缀和 mod k 的余数及其出现次数。
 * 3. 如果当前前缀和的 mod k 余数在 HashMap 中已经存在，说明从上一次出现该余数的位置到当前的位置之间的子数组之和可被 k 整除。
 * 4. 使用 Java Stream API 遍历数组，更新 HashMap 和结果计数。
 */

import java.util.HashMap;
import java.util.stream.IntStream;

public class Solution {
    public int subarraysDivByK(int[] nums, int k) {
        HashMap<Integer, Integer> map = new HashMap<>();
        map.put(0, 1);
        int[] count = {0};
        int[] prefixSum = {0};
        IntStream.of(nums).forEach(num -> {
            prefixSum[0] += num;
            int mod = (prefixSum[0] % k + k) % k; // 确保 mod 为正值
            count[0] += map.getOrDefault(mod, 0);
            map.put(mod, map.getOrDefault(mod, 0) + 1);
        });
        return count[0];
    }
}