Cod sursa(job #467350)

Utilizator darrenRares Buhai darren Data 28 iunie 2010 14:40:14
Problema Divk Scor 40
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 >= 0)     --v[s[i - b]]; 
		if (i - a + 1 >= 0) ++v[s[i - a + 1]];
		tot += v[s[i]];
	}
	fout << tot;
	return 0;
}