k倍区间-蓝桥杯

标题描述：k倍区间-蓝桥杯

#问题描述：

输入格式：

示例1：

输入：

5 2
1
2
3
4
5

输出：

6

提示

• 1 <= n,m <= 100
• 0 <= k <= 20

思路：

考虑前i个数的和对k求余，把余数相同的列进行累加，因为余数相同的列组成的区间相减所得的区间都可以构成k倍区间。
当新求得余数为 i 的列，可以与之前求得余数也为 i 的区间的数量 n 组成n个 k 倍区间。（余数不等于0则+n，余数等于0则本身也是k倍区间，所以+n+1）

AC_CodeByJava

import java.util.Scanner;
// 1:无需package
// 2: 类名必须Main, 不可修改

//考虑前i个数的和对k求余，根据相同余数，把最后求和的元素排一列。
//考虑余数相同的列组成的区间相减都可以构成k倍区间。
//（余数不等于0则+1，余数等于0则本身也是k倍区间，所以+2）
public class Main {
    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        //在此输入您的代码...
        int n = scan.nextInt();
        int k = scan.nextInt();
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) {
            arr[i] = scan.nextInt();
        }
        long count = 0;//k倍区间个数
        long sum = 0;//前i个数的和
        int[] modNum = new int[k];//如k=2, 余数有0 1。  modNum[i]存余数为i的元素个数
        //modNum[0]=1;
        for (int i = 0; i < arr.length; i++) {
            sum += arr[i];
            int mod = (int) (sum % k);
            if(mod==0){
                count++;//如果余数为0，本事就是k倍区间
            }
            //先累加原modNum[mod]，再modNum[mod]++
            count += modNum[mod];
            modNum[mod]++;
        }
        System.out.println(count);
        scan.close();
    }
}

时间复杂度分析