Cod sursa(job #2729626)

Utilizator SebicaPGLSebastian Ionel SebicaPGL Data 24 martie 2021 23:04:44
Problema Deque Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.04 kb
#include <bits/stdc++.h>
#include <fstream>

using namespace std;

int v[5000001];

int main()
{
    int n, k, inceput = 1, sfarsit = 0;
    deque<int> dequeue;
    long long 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 / cat timp elementul curent este mai mic decat ultimul din deck, atunci ii dam un pop_back dar fara STL
        while(!dequeue.empty() && v[i] < v[dequeue.back()]){
            dequeue.pop_back();
        }
        // dupa care push_back la pozitie
        dequeue.push_back(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.front() == i - k) dequeue.pop_front();

        // afisezi minimul
        if(i >= k - 1) sum += v[dequeue.front()];
    }

    o << sum;
    return 0;
}