Cod sursa(job #2249394)

Utilizator MaxTeoTeo Oprescu MaxTeo Data 29 septembrie 2018 19:10:33
Problema Divk Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.61 kb
#include <bits/stdc++.h>
using namespace std;

FILE *f = freopen( "divk.in", "r", stdin);
FILE *g = freopen( "divk.out", "w", stdout);

int n, k, a, b;
unsigned long long v[100005], ans;
int vf[500005];

int main()
{
    int x;
    scanf( "%d%d%d%d", &n, &k, &a, &b );
    for( int i = 1; i <= n; ++i)
    {
        scanf( "%d", &x );
        v[i] = v[i - 1] + x;
        if( i - a >= 0 )
            ++vf[ v[i - a] % k ];
        if( i - b -1 >= 0 )
            --vf[ v[i - b - 1] % k];
        if( i - a >= 0)
            ans += vf[ v[i] % k ];
    }
    printf( "%d\n", ans );
    return 0;
}