K倍区间

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

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

输入：
第一行包含两个整数N和K。(1 <= N, K <= 100000)
以下N行每行包含一个整数Ai。(1 <= Ai <= 100000)

输出：

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

例如，
输入：
5 2
1
2
3
4
5

程序应该输出：
6

资源约定：
峰值内存消耗（含虚拟机） < 256M
CPU消耗 < 2000ms

请严格按要求输出，不要画蛇添足地打印类似：“请您输入…” 的多余内容。
不要使用package语句。不要使用jdk1.7及以上版本的特性。
主类的名字必须是：Main，否则按无效代码处理。

起初还在想只要用for循环枚举i、j，把前面的几个的和去模k结果为0，cnt++就好，但是，这样写虽然答案没问题，可是时间复杂度很糟糕。

for (int i = 0; i <n; i++) {
	sum=0;
	for (int j = i; j < n; j++) {
		sum+=a[j];
		if (sum%k==0) {
			ans++;
		}
	}
}

对于区间和的维护，一般有三种方法：前缀和，树状数组，区间树。对于静态数据用前缀和就可以。

a[]   1 2 3 4 5 

b[] 0 1 3 6 10 15  这样就一下子把每一个前缀都得到了。

想要求某一个区间的和就只要把某一个前缀减另一个前缀就好。比如[2,5](a[]的值，并非下标)==>s[5]-s[2]

import java.util.Scanner;
public class Main{
    static int n,k;
    static int[] a;
    static int[] s;
    public static void main(String[] args){
        Scanner in=new Scanner(SYstem.in);
        n=in.nextInt();
        k=in.nextInt();
        a=new int[n+1];
        s=new int[n+1];
        s[0]=0;
        for(int i=1;i<=n;i++){
            a[i]=in.nextInt();
            s[i]=s[i-1]+a[i];
        }
        long ans=0;
        //枚举i,j
        for(int i=1;i<=n;i++){
            for(int j=i;i<=n;j++){
                //ij之间的区间和
                if((s[j]-s[i-1])%k==0){ans++;}
            }
        }
        System.out.println(ans);
    }
}

这样的优化虽然不错,但是数据大于10^4时还是会超时.

我们可以利用余数,s[]中的值每个都模k,余数为0的符号条件,而余数不为0的:两个相同余数的元素。他们的差一定为 k倍区间，{13%2=1 ;25%2=1 ;（25-13）%2==0}.所有任何两给余数非0且相同的数任意组合的次数为可以组成k倍区间个数.相当于

import java.util.Scanner;
public class Main {
    static int n,k;
    static int[] a;
    static int[] s;
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        n = in.nextInt();
        k = in.nextInt();
        a= new int[n + 1];
        s= new int[n + 1];
        long[] v=new long[200000];
        s[0]=0;
        for (int i = 1; i <=n; i++){
            a[i]=in.nextInt();
            s[i]=s[i-1]+a[i];
            s[i]%=k;
            v[s[i]]++;
            
        }
        long ans = 0;
        ans+=v[0];
        for (int i = 0; i <k; i++){//余数不会>=k
            ans+=(v[i]*(v[i]-1))/2;
        }
        System.out.println(ans);
    }
}