Cod sursa(job #1328665)

Utilizator alexmisto342Turdean Alexandru alexmisto342 Data 28 ianuarie 2015 17:27:17
Problema Divk Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.74 kb
#include <iostream>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <cstring>
using namespace std;
ifstream fin("divk.in");
ofstream fout("divk.out");
int v[500001],r[100000],i,j,a,b,n,k,x,y,start,capat;
long long int sol;
int main()
{
fin>>n>>k>>a>>b;x=0;
for(i=1;i<a;i++)
    {fin>>v[i];
    v[i]+=v[i-1];
    v[i]=v[i]%k;
    }
r[0]=1;
start=0;
capat=0;
for(i;i<=n;i++)
{
    fin>>v[i];
    v[i]+=v[i-1];
    v[i]=v[i]%k;
    sol+=r[v[i]];
    if(capat-start+1>b-a)
    {
        r[v[start]]--;
        start++;
        capat++;
        r[v[capat]]++;
    }
    else
        {
            capat++;
            r[v[capat]]++;
        }

}
fout<<sol;
fin.close();
fout.close();
return 0;
}