Cod sursa(job #1922909)

Utilizator ReeeBontea Mihai Reee Data 10 martie 2017 19:34:39
Problema Deque Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.1 kb
#include <fstream>

#define maxn 5000010

int N, K; /// vector de N elemente , secvente de K numere
int A[maxn], Deque[maxn];
int Front = 1 , Back = 0; /// Initializare, Back < Front => deque-ul este vid
long long Sum;

void Citire()
{
    ifstream fin ("deque.in");
    fin >> N  >> K;
    for ( int i = 1 ; i <= N ; ++i )
        fin >> A[i];
    fin.close();
}

int main()
{
    ofstream fout ("deque.out");
    Citire();
    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;
}