Cod sursa(job #1141662)

Utilizator barbuionBarbu Ion barbuion Data 13 martie 2014 00:41:25
Problema Divk Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.72 kb
#include<fstream>
#include<vector>
using namespace std;
long long nr,x,y,n,k,a,b,i,j,s;
int v;
vector <int> r[100011];
ifstream f("divk.in");
ofstream g("divk.out");
long long sol(int b){
long long nr=0,s=0;int st;
for(int i=0;i<k;i++){
        int st=0;
    for(int j=0;j<r[i].size();j++){

            while(st<=j&&r[i][j]-r[i][st]>b)
                {
                    st++;
                }
            if(st<=j)
                    nr+=j-st;
        }
}
for(int i=0;i<r[0].size()&&r[0][i]<=b;i++)
      nr++;
return nr;
}

int main(){



f>>n>>k>>a>>b;
v=0;
for(i=1;i<=n;i++){
f>>j;
v=(v+j)%k;
r[v].push_back(i);
}



nr=sol(b);
long long nr1=sol(a-1);


g<<nr-nr1;
return 0;
}