Cod sursa(job #467351)

Utilizator darrenRares Buhai darren Data 28 iunie 2010 14:41:39
Problema Divk Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.49 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;
		//v[r] = de cate ori apare restul 'r' in s[i - b + 1]...s[i - a + 1]
		
		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;
}