Cod sursa(job #905473)

Utilizator h2g2Ford Prefect h2g2 Data 5 martie 2013 21:09:33
Problema Divk Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.98 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <algorithm>
#include <deque>
#include <stack>
#define nmax 500005
#define kmax 100005
using namespace std;

int v[nmax], cu_rest[nmax], s[nmax];
int n, k, i, j, a, b, sol = 0;

int main() {
	ifstream f("divk.in");
	ofstream g("divk.out");

	f>>n>>k>>a>>b;

	s[0] = 0;
	for(i=1; i<=n; i++) {
        f>>v[i];
        s[i] = (s[i-1] + v[i])%k;
	}


    for(i=1; i-a+1 <= n; i++) {
        //in vectorul de resturi, adun 1 la restul s[]-ului in raport cu care formez secventa de lungime minima (a)
        //si scad 1 din numarul de resturi corespunzator s[]-ului in raport cu care formez secventa maxima (b)
        //ii un fel de numarare de resturi pe intervale de lungime data (ca intr-un deque)

        if(i - a + 1 >= 0) cu_rest[s[i-a+1]%k]++;
        if(i - b + 1 >= 0) cu_rest[s[i-b+1]%k] = max(cu_rest[s[i-b+1]%k]-1, 0);

        sol += cu_rest[s[i]%k];
	}


	cout<<sol<<"\n";

	return 0;
}