Cod sursa(job #1462437)

Utilizator StarGold2Emanuel Nrx StarGold2 Data 18 iulie 2015 00:55:27
Problema Divk Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.5 kb
#include <cstdio>
#define DIM 500010
using namespace std;

int N, M, K, A, B;
int V[DIM], P[DIM], S[DIM];
long long sol;

int main(){
	
	freopen("divk.in" ,"r", stdin );
	freopen("divk.out","w", stdout);
	
	scanf("%d %d %d %d", &N, &K, &A, &B);
	
	for(int i = 1; i <= N; i ++){
		
		scanf("%d", &V[i]);
		P[i] = ( P[i-1] + V[i] ) % K;
		
		if(i >= A)
			S[ P[i-A-0] ] ++;
			
		if(i > B)
			S[ P[i-B-1] ] --;
			
		sol += S[ P[i] ];
	}
	
	printf("%lld\n", sol);
	
	fclose(stdin );
	fclose(stdout);
	
	return 0;
}