Cod sursa(job #1126174)

Utilizator raulstoinStoin Raul raulstoin Data 26 februarie 2014 21:41:12
Problema Divk Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.5 kb
#include<fstream>

#define NMAX 500005

using namespace std;

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

int n,k,a,b,A[NMAX],B[NMAX],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;
	for(int i=1;i<=n;i++)
	{
		if(i==a && v[i]==0)
			sol=1;
		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;
}