Cod sursa(job #3224128)

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

using namespace std;

deque <pair <int, int>> 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 elementul cel mai din fata este prea vechi, il aruncam la gunoi
        while (!dq.empty() and i - dq.front().second >= k) dq.pop_front();

        // 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().first >= x) dq.pop_back();

        // pusham in spate elementul x cu pozitia i
        dq.push_back({x, i});

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