Cod sursa(job #1777853)

Utilizator CiurezAndreiCiurez Marius-Andrei CiurezAndrei Data 12 octombrie 2016 22:37:25
Problema Divk Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.67 kb
#include <bits/stdc++.h>

using namespace std;

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

int v[500010], D[500010], S[500010];
int N, K, A, B;
long long solution;

int main()
{
    fin >> N >> K >> A >> B;

    for(int i = 1; i <= N; i ++)
    {
        fin >> v[i];
        v[i] %= K;
    }

    for(int i = 1; i <= N; i ++)
    {
        D[i] += (D[i - 1] + v[i]) % K;
    }

    for(int i = 1; i <= N; i ++)
    {
        if(i >= A)
        {
            S[ D[i - A] ] ++;
        }
        if(i > B)
        {
            S[ D[i - B - 1] ] --;
        }
        solution += S[ D[i] ];
    }

    fout << solution;

    return 0;
}