【K倍区间-蓝桥杯原题/取模+前缀和】

从左到右一排有N个盒子。左边第一个盒子里装的是糖果。

你将把糖果从一些连续的盒子里拿出来,均匀地分发给M个孩子。

在这种情况下,找出满足下列条件的对(l,r)的数目:

l和r都是整数,满足1≤l≤r≤N.Al+Al+1+…+Ar是M的倍数。

约束

·输入的所有值都是整数。

·1≤N≤105

·2≤≤109

Ai·1≤≤109

输入

输入来自标准输入,格式如下:

 

N M

A1 A2……一个

 

输出

打印满足条件的对(l,r)的数目。

注意,该数字可能不适合32位整数类型。

 

样例输入副本

3 - 2

4 1 5

样例输出副本

3.

提示

 

·(1,1)的和:4

·(1,2)的和:5

·(1,3)的和:10

·(2,2)的和:1

·(2,3)的和:6

·(3,3)的和:5

其中3个是2的倍数。

前缀和+组合数学

 

由:(Sum[ R ]- Sum[ L-1 ])%K==0;——》Sum[ R ]%K -Sum [ L - 1 ] %K==0

 

把该数列的每一项的前缀和%m,如果在序列中有n个前缀和(%m后的)都等于p(p!=0)则说明在这p段中选出两端就行,为C_{n}^{2}

特别注意当p==0,为C_{n}^{2}+p.//p可以自成一段

首先我们知道表达一个区域『L,R』一定是用前缀和来处理对吧。
这个前缀和:sum[ R ]-sum[ L-1 ]。
我们想求一下K的倍数对吧。不就是对这个区域取模。
(Sum[ R ]- Sum[ L-1 ])%K==0;
但是我们知道,Sum[ R ]%K -Sum [ L - 1 ] %K==0 这个式子是等价对吧。
那么我们找的就是 (Sum[ L ]%K==0)或者(Sum[ R ] - Sum [ L-1 ])%K==0.
 
我们在任意一个   Sum【D】任意取两个作为两个端点就是,组合数:C_{n}^{2}
#include<iostream>
#include<algorithm>
#include<map>
#include <math.h> 
#include<memory.h>
using namespace std; 
typedef long long ll; 
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
const int maxn=5e5+100;
int a[maxn];
int sum[maxn];
map<int,int>mp;
int n,m; 
void inint(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        sum[i]=(sum[i-1]+a[i])%m;
        mp[sum[i]]++;
    }
}
int main(){
    inint();
    ll sum=mp[0];
    for(auto i=mp.begin();i!=mp.end();i++){
        ll s=i->second;
        sum+=(s*(s-1))/2;
    }
    printf("%lld",sum);    
}

 

posted @ 2020-03-13 22:53  lipu123  阅读(141)  评论(0编辑  收藏  举报