Cod sursa(job #701672)

Utilizator ms-ninjacristescu liviu ms-ninja Data 1 martie 2012 17:09:14
Problema Divk Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.51 kb
#include <fstream>
using namespace std;
#define dim 500005
int frec[dim], mat[dim];

int main()
{
	ifstream fin("divk.in");
	ofstream fout("divk.out");
	int n, k, a, b, i , j,nr;
	
	fin>>n >>k >>a >>b;
	int contor=0;
	for(i=1;i<=n;++i)
	{
		fin>>nr;
		mat[i]=(nr+mat[i-1])%k;
		if(mat[i]==0 && i>=a && i<=b)
			++contor;
	}
	
	for(j=a;j<=n;++j)
	{
		if(j>a)
			++frec[mat[j-a]];
		if(j>b+1)
			--frec[mat[j-b-1]];
			
		contor+=frec[mat[j]];
	}
	
	
	fout<<contor;
				
	
	return 0;
}