Cod sursa(job #3315215)
| Utilizator | Data | 13 octombrie 2025 01:27:51 | |
|---|---|---|---|
| Problema | Divk | Scor | 10 |
| Compilator | cpp-64 | Status | done |
| Runda | Arhiva de probleme | Marime | 0.54 kb |
#include <bits/stdc++.h>
using namespace std;
typedef long long i8;
int main() {
ifstream cin("divk.in");
ofstream cout("divk.out");
int n, k, a, b;
cin >> n >> k >> a >> b;
vector<int> pfx(n + 1, 0);
for (int i = 1; i <= n; i++) {
cin >> pfx[i];
pfx[i] += pfx[i - 1];
pfx[i] %= k;
}
vector<int> rest(k, 0);
int res = 0;
for (int i = a; i <= n; i++) {
int x = i - a;
int y = i - b;
if (y > 0) {
rest[pfx[y]]--;
}
rest[pfx[x]]++;
res += rest[pfx[i]];
}
cout << res << "\n";
return 0;
}