Cod sursa(job #3224167)

Utilizator Wolf98Vlad Munteanu Wolf98 Data 14 aprilie 2024 20:57:27
Problema Deque Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.1 kb
#include <bits/stdc++.h>

using namespace std;

struct pereche {
    int value, index;
};

deque <pereche> dq;

int main() {
    ifstream fin ("deque.in");
    ofstream fout ("deque.out");
    int n, k;
    long long sum = 0;
    fin >> n >> k;
    for (int i = 1; i <= n; ++i) {
        int x;
        fin >> x;

        // cat timp elemetul din spate este mai mare decat elementul curent, NU mai
        // avem nevoie de el pentru ca elementul curent este main nou decat el si este si mai mic
        // deci elementul din spate nu va mai putea deveni niciodata minim
        while (!dq.empty() and dq.back().value >= x) dq.pop_back();

        // cat timp elementul cel mai din fata este prea vechi, il aruncam la gunoi
        while (!dq.empty() and i - dq.front().index >= k) dq.pop_front();

        // pusham in spate elementul x cu pozitia i
        pereche current;
        current.value = x;
        current.index =  i;
        dq.push_back(current);

        if (i >= k) {
            // daca avem lungime k, adunam minimu, la suma
            sum += dq.front().value;
        }
    }
    fout << sum;
    return 0;
}