Cod sursa(job #3242738)

Utilizator gugalcromMuntoiu Vlad-Ioan gugalcrom Data 13 septembrie 2024 22:05:02
Problema Deque Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1 kb
#include <iostream>
#include <fstream>
#include <deque>
#include <vector>

using namespace std;

int main() {
    ifstream fin("deque.in");
    ofstream fout("deque.out");
    deque<pair<int, int>> coada;
    long long int suma=0, N, K, i;
    fin >> N >> K;
    vector<int> a(N+1);
    for(i=1; i<=N; ++i) {
        fin >> a[i];
    }
    for(i=1; i<=K; ++i) {
        while(!coada.empty() && coada.back().first >= a[i]) {
            coada.pop_back();    // ștergem elementele mai mari -- întotdeauna cel din spate este maxim
        }
        coada.push_back({a[i], i});
    }

    suma += coada.front().first; // minimul

    for(i=K+1; i<=N; ++i) {
        while(!coada.empty() && coada.back().first >= a[i]) {
            coada.pop_back();    // ștergem elementele mari
        }
        coada.push_back({a[i], i});
        while(!coada.empty() && coada.front().second <= i-K) {
            coada.pop_front();   // ștergem elementele vechi
        }
        suma += coada.front().first;
    }
    fout << suma;
}