Cod sursa(job #849069)

Utilizator enedumitruene dumitru enedumitru Data 6 ianuarie 2013 10:03:49
Problema Deque Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.03 kb
#include <cstdio>
#define Nmax 5000010
using namespace std;
int n, k, Front, Back, A[Nmax], Deque[Nmax];
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 ", &A[i]);
    Front = 1, Back = 0; // Initializare, 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 && A[i] <= A[ Deque[Back] ]) Back--;
        Deque[++Back] = i; // Adaugam pozitia elementului curent in deque
        // Daca elementul minim aflat in varful deque-ului 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++;
        if (i >= k) Sum += A[ Deque[Front]];      // insumam minimul, care se afla in varful deque-ului
    }
    printf("%lld\n", Sum);
    return 0;
}