Cod sursa(job #2876658)

Utilizator Stefan_BerlinschiStefan-Cristian Berlinschi Stefan_Berlinschi Data 23 martie 2022 13:24:24
Problema Deque Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.09 kb
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<vector>
#include<set>
#include<map>
#include<deque>
#include<fstream>

 
using namespace std;
 
#define x first
#define y second
#define NMAX 5000004

ifstream fin("deque.in");
ofstream fout("deque.out");
 
int v[NMAX];

deque<int> deq;
int n, k;
long long answer;
 
int main () {
 
    // 1. Primul element din deque o sa fie mereu minimul
    // 2. Toate elementele din deque sunt in ordine crescatoare
    // 3. Extragem elemente din ambele capete si inseram doar in capatul dreapta
 
    // scanf("%d%d", &n, &k);
    // for(int i = 1; i <= n; i++) {
    //     scanf("%d", &v[i]);
    // }
    fin >> n >> k;
    for (int i = 1; i <= n; i ++) {
        fin >> v[i];
    }
 
    for(int i = 1; i <= n; i++) {
        while (deq.empty() == false && v[i] <= v[deq.back()]) {
             deq.pop_back();
        }

        deq.push_back(i);


        if(i >= k) {
            answer += v[deq.front()];
            if(deq.front() == i - k + 1) {
                deq.pop_front();
            }
        }
    }
 
    fout << answer;
 
    return 0;
}