Cod sursa(job #2503513)

Utilizator FrostfireMagirescu Tudor Frostfire Data 3 decembrie 2019 12:24:29
Problema Divk Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.81 kb
#include <fstream>
#include <iostream>
#include <vector>
#define NMAX 500000

using namespace std;

ifstream f("divk.in");
ofstream g("divk.out");

int n, k, a, b;
long long sum, sol;
vector <int> v[NMAX+10];

void solve(int r)
{   int p1 = 0, p2 = 0, val;
    for(int i=1; i<v[r].size(); i++)
        {   val = v[r][i] - b;
            while(v[r][p1] < val) p1++;
            val = v[r][i] - a + 1;
            while(v[r][p2] < val) p2++;
            sol = sol + (long long)p2 - (long long)p1;
        }
}

int main()
{
    f >> n >> k >> a >> b;
    v[0].push_back(0);
    for(int i=1; i<=n; i++)
        {   long long x;
            f >> x;
            sum += x;
            v[sum%k].push_back(i);
        }
    for(int i=0; i<n; i++) solve(i);
    g << sol << '\n';
    return 0;
}