Cod sursa(job #1541428)

Utilizator raluca1234Tudor Raluca raluca1234 Data 4 decembrie 2015 00:42:43
Problema Deque Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.12 kb
//infoarena deque
#include<iostream>
#include<fstream>
#define maxn 5000010
using namespace std;

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

int main()
{
    ifstream f("deque.in");
    ofstream g("deque.out");
    int i;
    f>>N>>K;
    // Citesc sirul dat
    for (i=1; i<=N; i++)
        f>>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
        Back++;
        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]];
    }

    g<<sum;
    f.close();
    g.close();
    return 0;
}