Cod sursa(job #212787)

Utilizator alexeiIacob Radu alexei Data 6 octombrie 2008 20:48:44
Problema Divk Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.71 kb
#include<stdio.h>
#define NMAX 500024
#define KMAX 100024
long long S[NMAX],R[KMAX];
//Asta asa, ca sa nu invat pt testul de la mate:D
int main()
{
    freopen("divk.in","r",stdin);
    freopen("divk.out","w",stdout);
    
    int N,K,A,B;
    scanf("%d%d%d%d",&N,&K,&A,&B);
    
    int i;
    long long a1,solfin=0;
    for(i=1; i<=N; ++i){
    scanf("%lld",&a1);
    S[i]=S[i-1]+a1;
    }
    for(i=A; i<=B && i<=N; ++i)
    ++R[S[i]%K]; 
    
    for(i=2; i+A<=N; ++i)
    {
             solfin+=R[S[i-1]%K];
             if( i+B<=N )
                 ++R[ S[i+B-1]%K ];
             --R[ S[i+A-2]%K ];
    }
    solfin+=R[S[i-2]%K];
    printf("%lld\n",solfin);         
    return 0;
}