蓝桥杯——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） 

 

输出

输出一个整数，代表ķ倍区间的数目。 

 

样例输入

1

2

3

4

5

 

 

样例输出

6

学习过这道题让自己非常深刻地体会到了自己对前缀和知识的欠缺，天真的以为暴力就能解决一切

下次做题时一定要对连续区间要敏感，前缀和是一种多么方便的方法啊！！

 

看过网上各位大佬的题解之后，也对这个前缀和还有取模又有了一种新的理解

 

比如  [ l , r ] 上的区间判断其对k取模是否为0，用sum[]数组存储前缀和，即 [ l ,r ]上的区间和就等于sum [ r ] - sum [ l-1] 

而判断sum [ r ] - sum [ l-1] 对k取模是否为0,就是判断这两个前缀和对k取模是否相等

//  因为两个数对k取余后余数相同，相减后 他们分别对k 取余的余数就减掉了

//   即（sum [ r ] - sum [ l-1] %k ==0 可以 推导出sum [ r ] %k==sum [ l-1 ]%k

而要实现上面说的这些步骤，就是对前缀和取余k

例如题目给的 1，2，3，4，5

取模2后的前缀和分别是 1,1,0,0,1

其中 1 有三个 ，0 有两个，对所有的1或所有的0中，根据组合数 C(3,2) 或c(2,2)求出其中的取模2后相同的前缀和

 

代码解释：

//ans记录的是 不从第一个元素开始的区间的相同的个数，例如前面说的[2],[2,3,4,5],[3,4,5] (这些是取模2后为1的)，[4]（取模2后为0的）

//加上vis[0]是因为要记录从从第一个分别到第i个的区间中取模k 为 0的个数，例如（[1,2,3],[1,2,3,4])

代码如下

#include<stdio.h>
int main()
{
	int n,k,a[100007];
	    scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	long long sum[100007]={0};
	sum[1]=a[1]%k;
	for(int i=2;i<=n;i++)
		sum[i]=(sum[i-1]+a[i])%k;
	long long vis[100007]={0};    //    该数组是记录有多少个相同的区间和
	long long ans=0;
	for(int i=1;i<=n;i++){
		ans+=vis[sum[i]];
		vis[sum[i]]++;
	}
		printf("%lld",ans+vis[0]);    
        //ans记录的是 不从第一个元素开始的区间的相同的个数
        //加上vis[0]是因为要记录从从第一个分别到第i个的区间中取模k 为 0的个数
	return 0;
 }