Cod sursa(job #482902)

Utilizator mathboyDragos-Alin Rotaru mathboy Data 5 septembrie 2010 22:30:24
Problema Divk Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.46 kb
#include <cstdio>
#include <fstream>

#define maxn 500010
using namespace std;

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

int N, K, a, b, A[maxn], px, py, i, j;
long long sol;
int S[maxn], C[maxn];
int main ()
{

	fin >> N >> K >> a >> b;

	for (i = 1; i <= N; i++) {
		fin >> A[i];
		S[i] = (S[i - 1] + A[i]) % K;
	}
	++b;
	for (i = a; i <= N; i++) {
		
		++C[S[i - a]];
		if (i >= b) --C[S[i - b]];
		sol += C[S[i]];
	}
	fout << sol <<"\n";
	return 0;
}