Cod sursa(job #3314671)

Utilizator Cezar2009Cezar Mihai Titihazan Cezar2009 Data 10 octombrie 2025 17:36:58
Problema Divk Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.37 kb
//https://www.infoarena.ro/problema/divk

//#pragma GCC optimize ("Ofast")
//#pragma GCC optimize ("fast-math")
//#pragma GCC optimize ("unroll-loops")
//#define _USE_MATH_DEFINES

#include <iostream>
#include <fstream>
//#include <vector>
//#include <cstring>
//#include <cmath>
//#include <bitset>
//#include <queue>
//#include <stack>
//#include <utility>
//#include <algorithm>
//#include <string>
//#include <map>
//#include <unordered_map>
//#include <set>
//#include <unordered_set>
//#include <cstdint>
//#include <climits>
//#include <iomanip>
//#include <cstdio>
//#include <tuple>

using namespace std;

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

const int NRMAX = 500000;
const int KMAX = 100000;

int64_t sp[NRMAX + 1];
int fr[KMAX + 1];

int main()
{
    //ios_base::sync_with_stdio(false);
    //cin.tie(NULL);

    int n, k, a, b, i;
    int64_t rez = 0;

    fin >> n >> k >> a >> b;

    for (i = 1; i <= n; ++i)
    {
        int x;
        fin >> x;

        sp[i] = (sp[i - 1] + x) % k;

        // Pozitia pe care adaug i-a
        int adg = i - a;
        if (adg >= 0)
            ++fr[sp[i - a]];

        // Pozitia de pe care scot i-b-1
        int elim = i - b - 1;
        if (elim >= 0)
            --fr[sp[i - b - 1]];

        rez += fr[sp[i]];
    }

    fout << rez;

    return 0;
}