Cod sursa(job #2036242)

Utilizator petru.ciocirlanPetru Ciocirlan petru.ciocirlan Data 10 octombrie 2017 15:42:27
Problema Deque Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.92 kb
#include <fstream>
#include <deque>
#define maxn 5000010
using namespace std;
ifstream f("deque.in");
ofstream g("deque.out");
int n, k;
int A[maxn];
deque < int > D;
long long Sum;
int main()
{
    f>>n>>k;
    for (int i = 1; i <= n; i++) f>>A[i];

    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 (!D.empty() && A[i] <= A[ D.back() ]) D.pop_back();
        // Adaugam pozitia elementului curent in deque
        D.push_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 (D.front() == i-k) D.pop_front();
        // Afisam minimul, acesta aflandu-se in varful deque-ului
        if (i >= k) Sum += A[ D.front() ];
    }
    g<<Sum;
    g.close();
    return 0;
}