Cod sursa(job #3181)

Utilizator crawlerPuni Andrei Paul crawler Data 21 decembrie 2006 15:05:38
Problema Divk Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.5 kb
#include <stdio.h>

long tmp, sol, n, k, a, b, v[1<<19], r[1<<19], i;

int main()
{
  freopen("divk.in","r",stdin);
  freopen("divk.out","w",stdout);

  scanf("%ld %ld %ld %ld\n", &n, &k, &a, &b);

  for(i=1;i<=n;++i)
   {
     scanf("%ld\n",&tmp);
     r[i]=(r[i-1]+tmp)%k;
   }

  for(i=1;i<=a;++i)
   ++v[r[i]];
   
  sol+=v[r[b]];
  
  for(i=b+1;i<=n;++i)
   {
     sol+=v[r[i]];        
     --v[r[i-b]];
     ++v[r[i-a+1]];
   }

  printf("%ld\n", sol);

  return 0;
}