Cod sursa(job #2728864)

Utilizator SebicaPGLSebastian Ionel SebicaPGL Data 23 martie 2021 19:57:49
Problema Deque Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1 kb
#include <bits/stdc++.h>
#include <fstream>

using namespace std;

int v[5000001], dequeue[5000001];



int main()
{
    int n, k, inceput = 1, sfarsit = 0, sum = 0; // in etapa de initializare e 0, daca inc > sfarsit atunci deque e nul

    ifstream f("deque.in");
    ofstream o("deque.out");

    f >> n;
    f >> k;

    for(int i = 0; i < n; i++){
        f >> v[i];
    }
    for(int i = 0; i < n; i++){
        // daca elementul curent este mai mic decat ultimul din deck, atunci ii dam un pop_back dar fara STL
        while(inceput <= sfarsit && v[i] <= v[dequeue[sfarsit]]){
            sfarsit--;
        }
        // dupa care push_back la pozitie
        dequeue[++sfarsit] = i;

        // daca elementul minim coincide cu cel de pe pozitia i - k, ii eliminam pozitia din deque, ca nu mai conteaza pt >= i
        if(dequeue[inceput] == i - k) inceput++;

        // afisezi minimul
        if(i >= k) sum += v[dequeue[inceput]];
    }

    o << sum;
    return 0;
}