Cod sursa(job #2249397)
Utilizator | Teo Oprescu MaxTeo | Data | 29 septembrie 2018 19:15:25 |
---|---|---|---|
Problema | Divk | Scor | 50 |
Compilator | cpp | Status | done |
Runda | Arhiva de probleme | Marime | 0.55 kb |
#include <bits/stdc++.h>
#define ull unsigned long long
using namespace std;
ifstream f("divk.in");
ofstream g("divk.out");
int n, k, a, b;
ull v[100005], ans;
int vf[500005];
int main()
{
int x;
f >> n >> k >> a >> b;
for( int i = 1; i <= n; ++i)
{
f >> x;
v[i] = v[i - 1] + (ull)x;
if( i - a >= 0 )
++vf[ v[i - a] % k ];
if( i - b -1 >= 0 )
--vf[ v[i - b - 1] % k];
if( i - a >= 0)
ans += vf[ v[i] % k ];
}
g << ans;
return 0;
}