Cod sursa(job #2415265)

Utilizator ivan.tudorIvan Tudor ivan.tudor Data 25 aprilie 2019 17:59:11
Problema Divk Scor 40
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.65 kb
#include <iostream>
#include<cstdio>
using namespace std;
const int K=100005;
const int N=500005;
int frv[K];
int sp[N];
int main()
{
    FILE*fin,*fout;
    fin=fopen("divk.in","r");
    fout=fopen("divk.out","w");
    int n,a,b,k;
    fscanf(fin,"%d%d%d%d",&n,&k,&a,&b);
    for(int i=1;i<=n;i++){
      int x;
      fscanf(fin,"%d",&x);
      sp[i]=x+sp[i-1];
      sp[i]%=k;
    }
    frv[0]=1;
    long long sol=0;
    for(int i=a;i<=n;i++){
      frv[sp[i-a+1]]++;
      /*if(i==b+1)
        frv[0]--;*/
      if(i>=b)
        frv[sp[i-b]]--;
      sol+=1LL*frv[sp[i]];
    }
    fprintf(fout,"%lld",sol);
    return 0;
}