Cod sursa(job #2798972)

Utilizator andreea_chivuAndreea Chivu andreea_chivu Data 12 noiembrie 2021 10:05:07
Problema Deque Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.71 kb
#include <iostream>
#include <fstream>
#include <deque>

using namespace std;

#define NMAX 5000000

deque <int> d;
int v[NMAX + 1];

int main()
{
    ifstream fin("deque.in");
    ofstream fout("deque.out");

    int n, k;
    fin >> n >> k;

    for(int i = 0; i < n; i++){
        fin >> v[i];
    }

    long long s = 0;
    for(int i = 0; i < n; i++){
        if(!d.empty() && d.front() == i - k){
            d.pop_front();
        }
        while(!d.empty() && v[d.back()] >= v[i]){
            d.pop_back();
        }

        d.push_back(i);
        if(i >= k - 1){
            s+=v[d.front()];
        }
    }
    fout << s;
    fin.close();
    fout.close();
    return 0;
}