Cod sursa(job #3313)

Utilizator crawlerPuni Andrei Paul crawler Data 23 decembrie 2006 15:53:30
Problema Divk Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.53 kb
#include <stdio.h>

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

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

  register long i;

  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=a;i<=n;++i)
   {
      if(i>=a)
       ++v[r[i-a]];

      if(i>=b+1)
       --v[r[i-b-1]];
       
      sol+=v[r[i]];
   }

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

  return 0;
}
/*
 1 2 3 4 5 6
 2 1 1 0 1 1
*/