Cod sursa(job #1574831)

Utilizator Burbon13Burbon13 Burbon13 Data 20 ianuarie 2016 21:21:05
Problema Divk Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.59 kb
#include <cstdio>

using namespace std;

const int nmx = 500002;
const int kmx = 100002;

int n,k,a,b,v[kmx];
long long s[nmx],nr,sum;

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; i <= n; ++i){
        scanf("%lld", &nr);
        s[i] = s[i-1] + nr;
    }

    for(int i = 1; i <= n; ++i){
        if(i-a+1 > 0)
            ++ v[s[i-a+1]%k];
        if(i-b > 0)
            -- v[s[i-b]%k];
        sum += v[s[i]%k];
    }

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

    return 0;
}