Cod sursa(job #1127100)

Utilizator raulstoinStoin Raul raulstoin Data 27 februarie 2014 11:09:18
Problema Divk Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.51 kb
#include<fstream>

#define NMAX 500005
#define KMAX 100005

using namespace std;

ifstream fin("divk.in");
ofstream fout("divk.out");

int n,k,a,b,A[KMAX],B[KMAX],v[NMAX];
long long sol;

int main()
{
	fin>>n>>k>>a>>b;
	for(int i=1;i<=n;i++)
	{
		fin>>v[i];
		v[i]=(v[i]+v[i-1])%k;
	}
	A[0]=B[0]=1;
	if(!v[a] && a<=n)
		sol=1;
	for(int i=1;i<=n;i++)
	{
		if(i>a)
		{
			A[v[i-a]]++;
			sol+=A[v[i]];
		}
		if(i>b+1)
		{
			B[v[i-b-1]]++;
			sol-=B[v[i]];
		}
	}
	fout<<sol;
	return 0;
}