Cod sursa(job #467353)

Utilizator darrenRares Buhai darren Data 28 iunie 2010 14:44:45
Problema Divk Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.42 kb
#include<fstream>
using namespace std;

int a, b, n, k, s[500001], v[500001];
long long tot;

int main()
{
	ifstream fin("divk.in");
	ofstream fout("divk.out");
	fin >> n >> k >> a >> b;
	for (int i = 1, aux; i <= n; ++i)
	{
		fin >> aux;
		s[i] = (s[i - 1] + aux) % k;
		
		if (i - b - 1 >= 0) --v[s[i - b - 1]]; 
		if (i - a >= 0)     ++v[s[i - a]];
		tot += v[s[i]];
	}
	fout << tot;
	return 0;
}