Cod sursa(job #2232073)

Utilizator giotoPopescu Ioan gioto Data 17 august 2018 11:34:32
Problema Divk Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.55 kb
#include <bits/stdc++.h>
using namespace std;

int n, k, a, b;
int s[500005], f[100005];
int main()
{
    freopen("divk.in", "r", stdin);
    freopen("divk.out", "w", stdout);

    scanf("%d%d%d%d", &n, &k, &a, &b);
    int x;
    for(int i = 1; i <= n ; ++i){
        scanf("%d", &x);
        s[i] = (s[i - 1] + x) % k;
    }

    long long Sol = 0;
    ++f[0];
    for(int i = a; i <= n ; ++i){
        Sol = Sol + f[s[i]];
        ++f[s[i - a + 1]];
        if(i >= b) --f[s[i - b]];
    }

    printf("%lld", Sol);

    return 0;
}