Cod sursa(job #632675)

Utilizator mihaibogdan10Mihai Bogdan mihaibogdan10 Data 11 noiembrie 2011 23:31:24
Problema Deque Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.57 kb
#include<cstdio>
#include<deque>
using namespace std;
FILE *in = fopen("deque.in", "r"), *out = fopen("deque.out", "w");

deque <int> lista, indice;
int n, k;
long long s;

int main(){
	fscanf (in, "%d %d", &n, &k);
	int i, x;
	for (i = 1; i <= n; i++){
		fscanf(in, "%d", &x);
		while (!lista.empty() && x <= lista.back()) {lista.pop_back(); indice.pop_back();}
		lista.push_back(x);
		indice.push_back(i);
		if (indice.front() <= i - k) {lista.pop_front(); indice.pop_front();}
		if (i >= k) s += lista.front();
	}
	
	fprintf(out, "%lld", s);
	return 0;
}
#include <stdio.h>

#define maxn 5000010

int N, K;
int A[maxn], Deque[maxn];
int Front, Back;
long long Sum;

int main()
{
	freopen("deque.in", "r", stdin);
	freopen("deque.out", "w", stdout);

	int i;

	scanf("%d %d ", &N, &K);

	// Citesc sirul dat

	for (i = 1; i <= N; i++) 
		scanf("%d ", &A[i]);

	Front = 1, Back = 0; //	Initializare, Back < Front => deque-ul este vid

	for (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 && A[i] <= A[ 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 += A[ Deque[Front]]; 	
	}

	printf("%lld\n", Sum);

	return 0;
}