Cod sursa(job #2464608)

Utilizator darksky185Alexandru Gabriel darksky185 Data 28 septembrie 2019 17:45:30
Problema Divk Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.52 kb
#include <iostream>
#include <fstream>

using namespace std;

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

int v[500001];
long long s[500001];

int main()
{
    int n, k, a, b, i, x;
    long long r = 0;
    fin >> n >> k >> a >> b;
    for(i = 1; i <= n; ++i)
    {
        fin >> x;
        s[i] = (s[i-1] + x) % k;
        if(i >= a)
        {
            ++v[s[i-a]];
            if(i > b)
            {
                --v[s[i-b-1]];
            }
            r += v[s[i]];
        }
    }
    fout << r;

    return 0;
}