Cod sursa(job #2850116)

Utilizator Luca_CristianZamfir Luca-Cristian Luca_Cristian Data 16 februarie 2022 11:53:11
Problema Divk Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.57 kb
#include <iostream>
#include <fstream>

using namespace std;

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

#define MAXN 500001

int sum[MAXN], f[MAXN];
int main()
{

    int n, k , a, b, i, x;

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

    for(i = 1; i <= n; i++)
    {
        fin >> x;
        sum[i] = (sum[i - 1] + x) % k;
    }

    int j, rest;
    long long ans = 0;
    for(i = a; i <= n; i++)
    {
        f[sum[i - a]]++;
        if(i > b)
            f[sum[i - b - 1]]--;
        ans += f[sum[i]];
    }

    fout << ans;

    return 0;
}