Cod sursa(job #2469392)

Utilizator ruxiTTutunaru Ruxandra ruxiT Data 6 octombrie 2019 23:03:02
Problema Divk Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.53 kb
#include <iostream>
#include <fstream>

int v[500001];
int sum[500001];

using namespace std;

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

int main()
{
    int n, k, a, b, i, rez = 0, elem;
    fin >> n >> k >> a >> b;
    for (i = 1; i <= n; i++)
    {
        fin >> elem;
        sum[i] = (sum[i - 1] + elem) % k;
        if(i >= a)
        {
            v[sum[i-a]]++;
            if(i > b)
                v[sum[i-b-1]]--;
            rez += v[sum[i]];
        }
    }
    fout << rez;
    return 0;
}