Cod sursa(job #2631218)

Utilizator yzarayarina yzara Data 29 iunie 2020 15:00:16
Problema Divk Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.42 kb
#include <fstream>

using namespace std;

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

long long v[500005],f[500005],n,k,rez,a,b,nr;

int main()
{
    in>>n>>k>>a>>b;
    for ( int i=1; i<=n; i++)
    {
        in>>nr;
        v[i]=(v[i-1]+nr)%k;
        if ( i>=a)
            f[v[i-a]]++;
        if ( i>b)
            f[v[i-b-1]]--;
        rez=rez+f[v[i]];
    }
    out<<rez;
    return 0;
}