蓝桥 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




　　请严格按要求输出，不要画蛇添足地打印类似：“请您输入...” 的多余内容。


　　注意：
　　main函数需要返回0;
　　只使用ANSI C/ANSI C++ 标准;
　　不要调用依赖于编译环境或操作系统的特殊函数。
　　所有依赖的函数必须明确地在源文件中 #include <xxx>
　　不能通过工程设置而省略常用头文件。


　　提交程序时，注意选择所期望的语言类型和编译器类型。

思路 ： 抽屉原理：

【抽屉原理】：原理1 把多于n个的物体放到n个抽屉里，则至少有一个抽屉里的东西不少于两件；

                     原理2 把多于mn(m乘以n)个的物体放到n个抽屉里，则至少有一个抽屉里有不少于m+1的物体。

                     原理3 把无穷多件物体放入n个抽屉，则至少有一个抽屉里 有无穷个物体。.

                     原理 4 把（mn－1）个物体放入n个抽屉中，其中必有一个抽屉中至多有（m—1）个物体。

求出前缀和 sum[i]   int x= sum[i]%k   我们将x  放入抽屉  这里有k 个抽屉  分别为 【0，k-1】  当然抽屉原理 有解的情况一定是 n>k   但是本题结果也可能为0   。   

假设 cnt【x】==2   那么表明 在前缀和中  一定有sum[i] 和sum[j]  %k ==x   那么我们就一定可以确定 从[i+1, j] 这一段的和 一定为 k 的倍数 

假设  cnt【x】==3   那么就一定有  sum[i]  sum[j]  sum[k] %k==x  那么 从[i+1,j]  [j+1,k]  [i+1,k] 的和 也一定为k的倍数。

代码：

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#define N 100005

using namespace std;
typedef long long ll; 

ll sum[N];
ll k;
int n;
ll cnt[N];

int main()
{
	int x;
	cin>>n>>k;
	for(int i=1;i<=n;i++)
	{
		cin>>x;
		sum[i]=sum[i-1]+x;
	}
	
	cnt[0]++;//sum【0】 可以放入 0 的抽屉
	for(int i=1;i<=n;i++)
	{
		cnt[sum[i]%k]++;
	}
	
	ll ans=0;
	for(int i=0;i<k;i++){
		ans+=(cnt[i]*(cnt[i]-1))/2;
	}
	
	printf("%lld\n",ans);
	
	return 0;
}