蓝桥杯-历届试题 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

---

解题思路：
1、暴力搜索，不过这种方法明显不是我们要的
2、前缀和优化计数，如题解

import java.util.*;

public class Main{  
		
	static Scanner sc = new Scanner(System.in);
	
    public static void main(String[] args) {
    	int n,m;
    	n = sc.nextInt();
    	m = sc.nextInt();
    	int[] sum = new int[n+1];
    	int[] cnt = new int[n+1];
    	int a;
    	for(int i=1;i<=n;i++){
    		a = sc.nextInt();
    		sum[i] = (sum[i-1]+a)%m; // 思路起飞第一步
    		System.out.print(sum[i]+" ");
    	}
    	System.out.println();
    	long ans=0;
    	// 思路起飞第二步
    	for(int i=1;i<=n;i++){
    		ans += cnt[sum[i]];
    		cnt[sum[i]] ++;
    		if(sum[i]==0)
    			ans++;
    	}
    	System.out.println(ans);
	}
}

这里可能很多人都能理解第16行代码的意思，但是不知道为什么这么做，到了第22行的for循环就更加懵逼了，这也是学习算法题经常遇到的情况，明明每一行代码我都看得懂，但是就是不知道为什么要这么做，在这里我解释一下：
题目所求的是求区间[i，j]使得(sum[j]-s[i-1])%m=0，这个很多人都知道，也都可以根据这个编出对应的代码，但是直接根据这个公式编出来的程序时间复杂度大概是n^2，基本没什么难度，但也拿不到满分，那么我们把公式变形一下，(sum[j]-sum[i-1])%m=0 >>> sum[j]%m=sum[i-1]%m，也就是说如果区间[i，j]满足要求，那么就有sum[j]%m=sum[i-1]%m，到这里再回过头来看一下第23和24行代码相信大家应该可以理解了，最后，sum[i]%m=0是上述公式的一种特殊情况，也应该加上