第八届蓝桥杯JavaB组第十题：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

解析：
需要用到前缀和知识

什么是前缀和：
前缀和算法是一种重要的预处理算法，能大大降低查询的时间复杂度。最简单的题目就是：给定n个数和m次询问，每次询问一段区间的和。
查找一个区间的和我们可以从加法转换为减法，查询从L到R区间数字的和，其实可以转换为前R之和减去前L-1项之和。如下所示：

sum[0]=0;
for(i=1;i<=n;i++)
{

      scanf("%d",&a[i]);

       sum[i]=sum[i-1]+a[i];

} 

数据较大所以用long

因为K的范围100000，双循环简单解决但是会超时，所以想办法用单循环

我们先来举一个简单的例子：在数组区间[i,j]中，我们可以设i<=k<=j,设数组前k个元素的和为sum[k],前j个元素的和和为sum[j],这是，假设如果sum[k]%K == sum[j]%K的时候，我们可以发现，（sum[j] - sum[k]）% K ==0 ,由此我们可以看出，在区间[k , j]中的元素之和是K的倍数，所以我们可以利用这个特点来统计在数组[i , j]中，有相同余数的子序列之和的个数，即cnt[sum[k]%K],便可进行进一步的统计操作，举一个简单的例子
　　数列 1 2 3 4 5 mod = 2
　　对前1个数的和取模， 为1 之前有0个前缀和取模后为1，个数+0
　　对前2个数的和取模， 为1 之前有1个前缀和取模后为1，个数+1
　　对前3个数的和取模， 为0 之前有0个前缀和取模后为0, 个数+0
　　对前4个数的和取模， 为0 之前有1个前缀和取模后为0，个数+1
　　对钱5个数的和取模， 为1 之前有2个前缀和取模后为1，个数+2
　　到目前为止ans = 4。但是ans应该等于6，因为这样计算后，我们漏掉了前i个数的和取模是k的倍数的情况，即[0,i]区间和是k的倍数，因此，我们要在ans = 4 的基础上 加上前缀和取模后为0的个数 即ans+2 = 6;

import java.util.Scanner;

public class Main
{
    public static void main(String[] args)
    {
        Scanner in = new Scanner(System.in);
        int[]arr=new int[100001];
        int[]sum=new int[100001];
        int[]cnt=new int[100001];
        sum[0]=0;
        int n=in.nextInt();
        int k=in.nextInt();
        long ans=0;
        for(int i=1;i<=n;i++)
        {
            arr[i]=in.nextInt();
            //每个前缀和都%k
            sum[i]=(sum[i-1]+arr[i])%k;
            //用cnt[]数组记录每个%k后的值的个数
            //ans记录区间的个数
            ans+=cnt[sum[i]];
            cnt[sum[i]]++;
        }
        //最后还要加上%k后为0的个数
        System.out.println(ans+cnt[0]);
        //关闭输入，养好习惯
        in.close();
    }
}