（Java）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

思路分析

这题竟然蓝桥压轴，第一眼感觉挺水的，前缀和直接计算区间，也不需要判重
关于前缀和的描述可以看我之前的blog
点着跳转查看关于我的有关前缀和的blog
但是仔细一看，数据量为10的5次方，说明不能超过logn
如果按下面暴力的话，只能拿一点分

暴力代码实现

package practise.competition2017;

import java.util.ArrayList;
import java.util.Scanner;

public class k倍区间 {
    static int thesum=0;
    static int k=0;
   static ArrayList<Integer> list = new ArrayList<>();
    public static void main(String[] args) {


        Scanner scanner = new Scanner(System.in);
        int N=scanner.nextInt();
        int K=scanner.nextInt();
        k=K;

        int[] a = new int[N];
        for (int i = 0; i < N; i++) {
            a[i]=scanner.nextInt();
        }
        int[] sum = new int[N+1];
        for (int i = 0; i < N; i++) {
            sum[i+1]=sum[i]+a[i];
        }

        for (int i = 0; i < N; i++) {
            for (int j = i; j < N; j++) {
                if (i==j){
                    thesum=a[i];
                }else {
                    thesum=sum[j+1]-sum[i];
                }
                makenum(thesum);
            }
        }

        System.out.println(list.size());
    }

  public static void makenum(int sum){
        if(sum%k==0){
            list.add(sum);
        }
  }
}

之后仔细想了想，这道题想拿满分，还得靠之前学过的数论中的同余定理

满分思路及代码实现

现在来简单介绍一下什么是同余定理

这样我们在求区间的倍数时：

• (presum[j]-presum[i]）%k ==0 时
  就可以转化成
• presum[i] %k ==presum[j]%k

我们只要使用一个记忆化计数数组，将取余后的前缀和放在数组中，每次只要遍历前缀和数组，并累加上每个前缀和数组的值在记忆化数组的数量即可

• 特别注意的是每次前缀数组都要对其取余防止越界，不然只能拿一半分

import java.util.Scanner;

public class Main {
	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		int m=scanner.nextInt();
		int k=scanner.nextInt();
		int[] arr=new int[m];
		for (int i = 0; i < arr.length; i++) {
			arr[i]=scanner.nextInt();
		}
		// 存储数组,  (presum[j]-presum[i])%k==0; ==>  
		/*
		 * presum[j]%k   a
		 * presum[i]%k   b
		 * --> (presum[j]-presum[i])%k== a-b=0
		 * --> a==b
		 * 
		 * */
		int[] presum=new int[arr.length];
		presum[0]=arr[0]%k;
		for(int i=1;i<arr.length;i++) {
			presum[i]=(presum[i-1]+arr[i])%k;
		}
		// 1 3 6 10 15
		// 1 1 0 0  1
		// 0 1 0 1  2
		int[] cnt=new int[k];// 记录前缀和取余后的数量
		long count=0;
		for(int i=0;i<presum.length;i++) {
			count+=cnt[presum[i]]++; // 匹配和上次相同的
		}
		count+=cnt[0];
		System.out.println(count);
		
	}
}