Cod sursa(job #2243726)
Utilizator | Teo Oprescu MaxTeo | Data | 21 septembrie 2018 11:44:52 |
---|---|---|---|
Problema | Divk | Scor | 50 |
Compilator | cpp | Status | done |
Runda | Arhiva de probleme | Marime | 0.61 kb |
#include <bits/stdc++.h>
using namespace std;
FILE *f = freopen( "divk.in", "r", stdin);
FILE *g = freopen( "divk.out", "w", stdout);
long long n, k, a, b, ans;
long long v[100005];
int vf[500005];
int main()
{
int x;
scanf( "%d%d%d%d", &n, &k, &a, &b );
for( int i = 1; i <= n; ++i)
{
scanf( "%d", &x );
v[i] = v[i - 1] + 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 ];
}
printf( "%d\n", ans );
return 0;
}