Cod sursa(job #2447782)

Utilizator lucian2015blaugranadevil lucian2015 Data 14 august 2019 16:22:06
Problema Divk Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.42 kb
#include <fstream>
#define nmax 500001
#define nmax1 100001
using namespace std;


ifstream f("divk.in");
ofstream g("divk.out");


int main(){
	int n, k, a, b, x;
	long long sums[nmax]={0}, vec[nmax1]={0};
	int i, j;
	long long nrsecv=0;
	f>>n>>k>>a>>b;
	for(i=1;i<=n;i++){
		f>>x;
		sums[i]=(sums[i-1]+x)%k;
		if(i>=a)
			++vec[sums[i-a]];
		if(i>b) 
			--vec[sums[i-b-1]];
		nrsecv=nrsecv+vec[sums[i]];
	}
	g<<nrsecv<<" ";
}