Cod sursa(job #2396044)
Utilizator | Data | 3 aprilie 2019 10:26:42 | |
---|---|---|---|
Problema | Divk | Scor | 20 |
Compilator | cpp-64 | Status | done |
Runda | Arhiva de probleme | Marime | 0.68 kb |
#include <iostream>
#include <fstream>
using namespace std;
ifstream fin("divk.in");
ofstream fout("divk.out");
long long v[500005], d[100005];
int main()
{
int n, k, a, b, x;
long long s = 0;
fin >> n >> k >> a >> b;
for (int i = 1; i <= n; i++)
{
fin >> x;
v[i] = v[i - 1] + x;
}
int p = 0, q = 1;
for (int i = 1; i <= n; i++)
{
p++;
d[v[p] % k]++;
if (p - q > b - a)
{
d[v[q] % k]--;
q++;
}
if (p + a - 1 > n)
break;
else
s += d[v[p + a - 1] % k];
}
fout << s << '\n';
return 0;
}