Cod sursa(job #1959243)

Utilizator MiricaMateiMirica Matei MiricaMatei Data 9 aprilie 2017 11:32:05
Problema Divk Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.52 kb
#include <cstdio>
using namespace std;
int f[100005];
int v[500005];
int sp[500005];
int main(){
  freopen("divk.in", "r", stdin);
  freopen("divk.out", "w", stdout);
  int n, k, a, b;
  long long sol = 0;
  scanf("%d%d%d%d", &n, &k, &a, &b);
  for (int i = 1; i <= n; ++i){
    scanf("%d", &v[i]);
    v[i] %= k;
    sp[i] = (sp[i - 1] + v[i]) % k;
  }
  for (int i = a; i <= n; ++i){
    if (i > b)
      f[sp[i - b]]--;
    sol += f[sp[i]];
    f[sp[i]]++;
  }
  printf("%lld\n", sol);
  return 0;
}