蓝桥杯——K倍区间（JAVA）

题目：

给定一个长度为 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

运行限制

• 最大运行时间：2s
• 最大运行内存: 256M

思路：

（1）前缀数（从序列第一个数到它自身的所有数之和）：比如例题中的序列1 2 3 4 5的前缀数分别是1 3 6 10 15。

（2）前缀数相减得到的区间：两个前缀数相减会得到一个区间比如5的前缀数15减去2的前缀数3可得到一个区间3 4 5。

（3）储存同余的前缀数：将所有前缀数对K取余将每个余数及其个数进行存储如B[1]=3表示余数为1的前缀数有3个。

（4）两个同余的前缀数相减得到的区间一定是K倍区间：这算是一种数学规律如例题中15%2=1，3%2=1，（15-3）%2=0也就是3 4 5是一个K倍区间。

（5）如果前缀数对K取余为0则说明这一区间必为K倍区间：如10%2=0，则1 2 3 4必为一个K倍区间，

result=result+B[0];直接加上即可。

（6）根据（4）可知任何两个同余的前缀数都可以获得一个前缀区间，则进行排列组合根据同余前缀数个数求出结果即可，还以例题为例B[1]=3说明有三个余数是一的前缀数，我们要从中挑选两个组成一组求一个区间，共有C32（高中的排列组合3在下2在上）种组合方式为3*（3-1）/2=3，同理，余数为0的前缀（（5）计算的是从第一个数开始到它本身这个区间是k倍区间的个数所以并不冲突）数有1种组合方式加到一起总共有2+1=3种再加上（5）中的2种共有6种。

 

代码： 

import java.util.Scanner;


public class  Main{
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();
        int K = sc.nextInt();
        int[] A = new int[N];           //储存数列
        long[] B = new long[100000];    //储存同余的前缀和的个数
        long sum=0;                     //计算前缀和
        long result = 0;                //储存结果
        for (int i = 0; i < N; i++) {
            A[i]= sc.nextInt();
            sum+=A[i];
            B[(int)(sum % K)]++;        //前缀和取余并记数
        }
        result=result+B[0];             //余数为零的区间一定是k倍区间
        for(int i=0;i<K;i++){
            result+=(B[i]*(B[i]-1))/2;  //相同余数的前缀数相减得到的区间一定是K倍区间，再利用排列组合计算个数。
        }
        System.out.println(result);
        sc.close();            
    }
}