Cod sursa(job #1098801)

Utilizator pulseOvidiu Giorgi pulse Data 5 februarie 2014 11:09:12
Problema Deque Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.99 kb
#include <cstdio>

using namespace std;

#define NMAX 5000003

int n, k;
int deque[NMAX], v[NMAX];
int front, back;
long long sum;

int main ()
{
	freopen ("deque.in", "r", stdin);
	freopen ("deque.out", "w", stdout);
	scanf ("%d %d", &n, &k);
	for (int i = 1; i <= n; ++i)
		scanf ("%d", &v[i]);

	front = 1, back = 0; // initializare, daca back <= front deque-ul este vid

	for (int i = 1; i <= n; ++i)
	{
		// Cat timp elementul curent este mai mic decat ultimul element din deque, eliminam pozitia ultimului element din deque
		while (front <= back && v[i] <= v[deque[back]]) --back;
		// Adaugam pozitia elementului curent in deque
		deque[++back] = i;
		// Daca elementul minim coincide cu cel de pe pozita i-K, ii eliminam pozitia din deque, deoarece acesta nu mai conteaza pentru pasii >= i
		if (deque[front] == i - k) ++front;
		// Afisam minimul, acesta aflandu-se in varful deque-ului
		if (i >= k) sum += v[deque[front]];
	}

	printf ("%lld", sum);
	return 0;
}