Cod sursa(job #2576252)

Utilizator cristicretuCristi Cretu cristicretu Data 6 martie 2020 18:05:45
Problema Divk Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.45 kb
#include <fstream>
#define ll long long
using namespace std;
ifstream fin("divk.in");
ofstream fout("divk.out");
ll n, k , a, b, sol, v[500010], f[100010], x, i;
int main()
{
    fin >> n >> k >> a >> b;
	for(i = 1; i <= n; i++)
	{
		fin >> x;
		v[i] = ( v[i-1] + x )%k;
	}
	for(i = 1; i <= n; i++)
		{
			if( i - a >= 0)
			++f[v[i - a]];
			if( i - b > 0 )
				--f[v[i - b - 1]];

	sol += f[v[i]];
	}
	fout << sol;
    return 0;
}