Cod sursa(job #1461280)

Utilizator robx12lnLinca Robert robx12ln Data 15 iulie 2015 12:23:37
Problema Divk Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.49 kb
#include<fstream>
using namespace std;
ifstream fin("divk.in");
ofstream fout("divk.out");
int v[500005],st[100005];
int n,k,a,b,i;
long longsol;
int main(){
    fin>>n>>k>>a>>b;
    for(i=1;i<=n;i++){
        fin>>v[i];
        v[i]=v[i]+v[i-1];
    }
    for(i=1;i<=n;i++){
        v[i]%=k;
    }
    for(i=a;i<=n;i++){
        st[ v[i-a] ] ++;
        if(i>b){
            st[ v[i-b-1] ] --;
        }
        sol+=st[v[i]];
    }
    fout<<sol<<"\n";
    return 0;
}