Cod sursa(job #1073034)

Utilizator Mihai22eMihai Ionut Enache Mihai22e Data 5 ianuarie 2014 16:25:26
Problema Divk Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.56 kb
#include <cstdio>
using namespace std;

const int MAX_N = 500002;
const int MAX_K = 100002;

int N, K, A, B;
int s[MAX_N], cnt[MAX_K];
long long sol;

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

    scanf("%d %d %d %d", &N, &K, &A, &B);
    for(int i = 1, val; i <= N; ++i) {
        scanf("%d", &val);
        s[i] = (s[i - 1] + val) % K;
    }

    for(int i = A; i <= N; ++i) {
        ++cnt[s[i - A]];
        if(i > B)
            --cnt[s[i - B - 1]];
        sol += cnt[s[i]];
    }

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

    return 0;
}