Acwing 1230.K倍区间 前缀和 求余

给定一个长度为 N 的数列，A1,A2,…AN如果其中一段连续的子序列 Ai,Ai+1,…Aj之和是 K 的倍数，我们就称这个区间 [i,j] 是 K倍区间。

你能求出数列中总共有多少个 K倍区间吗？

输入格式

第一行包含两个整数 N和 K。

以下 N行每行包含一个整数 Ai。

输出格式

输出一个整数，代表 K倍区间的数目。

数据范围

1≤N,K≤100000
1≤Ai≤1000001

输入样例：

5 2
1
2
3
4
5

输出样例：

6

思路： 

首先此题是要求一个区间内元素的和是否是k倍区间，因此我们很容易得就能联想到前缀和。我们可以求出一个前缀和数组，便于后面的计算。求出这个前缀和数组后，开始的想法就是双指针遍历，看是否有符合要求的区间。但是这样是O(N^2)时间复杂度。会超时。因此我们需要优化。

一个区间 [i,j] 的和是k倍区间，说明前缀和(sumj - sumi) % k == 0。因此我们可以得出 sumi % k == sumj % k。由此，我们就可以根据余数是否相同，判断能组成多少个区间。我们维护一个cnt数组，cnt[i] 记录前缀和数组中对k求余 的余数为i 的元素个数。当sumi % k == p时同时cnt[p] == m时，我们就可以得知有m种组合方式使得sumj - sumi为k倍区间。

同时，由于余数为0的自身就是一个区间，因此我们要保持余数为0 的初始值为0.

代码:

import java.util.*;
public class Main{
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int[] cnt = new int[100010];
        int n = sc.nextInt();
        int k = sc.nextInt();
        int[] ori = new int[n];
        long[] sum = new long[n + 1];
        for (int i = 0; i < n; i++) {
            ori[i] = sc.nextInt();
            sum[i + 1] = sum[i] + ori[i];
        }
        cnt[0] = 1;
        long ans = 0;
        for (int i = 1; i < n + 1; i++) {
            ans += cnt[(int)(sum[i] % k)];
            cnt[(int)(sum[i] % k)]++;
        }
        System.out.println(ans);
    }
}