Cod sursa(job #1493302)

Utilizator dorin31Geman Dorin Andrei dorin31 Data 28 septembrie 2015 23:14:32
Problema Deque Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.99 kb
#include <fstream>
#define maxn 5000001

using namespace std;

int n, k, a[maxn], deque[maxn];
int front, back;
long long sum;

int main()
{
    ifstream fin("deque.in");
    ofstream fout("deque.out");
    fin>>n>>k;
    for (int i=1; i<=n; ++i)
        fin>>a[i];

    front=1; back=0; // Initializare. back<front -> deque 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--;
        // 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]];
    }

    fout<<sum;
    return 0;
}