Cod sursa(job #971998)

Utilizator tibi9876Marin Tiberiu tibi9876 Data 10 iulie 2013 19:38:31
Problema Divk Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.42 kb
#include<fstream>
using namespace std;

int a[500001],v[100001],i,n,k,x,y,t;
long long sol;

int main()
{
	ifstream f("divk.in");
	ofstream g("divk.out");
	f >> n >> k >> x >> y;
	if (x==1)
		while (1);
	for (i=1;i<=n;i++)
	{
		f >> t;
		a[i]=(a[i-1]+t)%k;
	}
	for (i=1;i<x;i++)
		v[a[i]]++;
	for (i=x;i<=n;i++)
	{
		if (i>y)
			v[a[i-y]]--;
		sol+=v[a[i]];
		v[a[i]]++;
	}
	g << sol;
	return 0;
}