Cod sursa(job #2249397)

Utilizator MaxTeoTeo Oprescu MaxTeo Data 29 septembrie 2018 19:15:25
Problema Divk Scor 50
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.55 kb
#include <bits/stdc++.h>
#define ull unsigned long long
using namespace std;

ifstream f("divk.in");
ofstream g("divk.out");

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

int main()
{
    int x;
    f >> n >> k >> a >> b;
    for( int i = 1; i <= n; ++i)
    {
        f >> x;
        v[i] = v[i - 1] + (ull)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 ];
    }
    g << ans;
    return 0;
}