Cod sursa(job #701237)

Utilizator paunmatei7FMI Paun Matei paunmatei7 Data 1 martie 2012 14:39:39
Problema Divk Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.44 kb
#include<stdio.h>
long long n,k,a,b,ap[100007],s[500007],v[500007],sum,i;
int main()
{
	freopen("divk.in","r",stdin);
	freopen("divk.out","w",stdout);
	scanf("%lld %lld %lld %lld",&n,&k,&a,&b);
	for (i=1;i<=n;i++)
		scanf("%lld",&v[i]);
	for (i=1;i<=n;i++)
		s[i]=(s[i-1]+v[i])%k;
	for(i=1;i<=n;i++)
    {
		if(i-a>=0)
          ++ap[s[i-a]];
		if(i-b>0)
          --ap[s[i-b-1]];
		sum+=ap[s[i]];
    }
	printf("%lld",sum);
}