Cod sursa(job #2850102)

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

using namespace std;

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

#define MAXN 500001

int sum[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, ans = 0;
    for(i = 1; i <= n - a + 1; i++)
    {

        j =  i + a - 1;
        while(j <= i + b - 1 && j <= n)
        {
            if(((sum[j] - sum[i - 1]) + 2 * k) % k == 0)
                ans++;
            j++;
        }
    }

    fout << ans;

    return 0;
}