Cod sursa(job #949272)

Utilizator enedumitruene dumitru enedumitru Data 13 mai 2013 06:03:41
Problema Divk Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.43 kb
#include <fstream>
#define nmax 500002
using namespace std;
ifstream f("divk.in"); ofstream g("divk.out");
int n,k,a,b,x;
long long nr;
int sume[nmax],sol[nmax];
int main()
{   f>>n>>k>>a>>b;
    for(int i=1; i<=n; i++)
		{f>>x; sume[i]=(sume[i-1]+x) % k;}
    for(int i=1; i<=n; i++)
	{   if(a<=i) ++sol[sume[i-a]];
        if(b<i) --sol[sume[i-b-1]];
        nr+=sol[sume[i]];
    }
    g<<nr<<"\n"; g.close(); return 0;
}