Cod sursa(job #1721660)

Utilizator liviu23Liviu Andrei liviu23 Data 26 iunie 2016 12:10:01
Problema Divk Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.53 kb
#include <stdio.h>
#include <iostream>
using namespace std;

int n,k,a,b,frecv[100005],sum[500005];

int main()
{
    freopen("divk.in","r",stdin);
    freopen("divk.out","w",stdout);
    scanf("%d%d%d%d",&n,&k,&a,&b);
    long long int rasp=0;
    int x;
    for(int i=1;i<=n;i++) {
        scanf("%d",&x);
        sum[i]=(sum[i-1]+x)%k;
        if(i>=a)
            frecv[sum[i-a]]++;
        if(i>b)
            frecv[sum[i-b-1]]--;
        rasp+=(1LL*frecv[sum[i]]);
    }
    printf("%lld",rasp);
    return 0;
}