Cod sursa(job #1561237)

Utilizator antanaAntonia Boca antana Data 3 ianuarie 2016 20:09:07
Problema Divk Scor 40
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.54 kb
#include <cstdio>
#include <vector>
#define MAX 100000
using namespace std;
int vc[MAX+1], v[500001];
int main()
{
    freopen("divk.in", "r", stdin);
    freopen("divk.out", "w", stdout);
    int n, k,a,b, i, s=0,  x;
    long long r=0;
    scanf("%d%d%d%d", &n, &k, &a, &b);
    for(i=1;i<=n;i++)
    {
        scanf("%d", &x);
        v[i]=(v[i-1]+x)%k;
    }
    for(i=a;i<=n;i++)
    {
        vc[v[i-a]]++;
        r=r+vc[v[i]];
        if(i>b)
            vc[v[i-b-1]]--;
    }
    printf("%lld", r);
    return 0;
}