Cod sursa(job #3327)

Utilizator crawlerPuni Andrei Paul crawler Data 23 decembrie 2006 16:10:23
Problema Divk Scor 40
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.58 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+2)
       --v[r[i-b-2]];

      sol+=v[r[i]];
   }

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

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

  2 9 4 5
  5 4 1
  4 1
  1 4
*/