Cod sursa(job #2876656)

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

 
using namespace std;
 
#define x first
#define y second
#define NMAX 5000004
 
int v[NMAX];

deque<int> deq;
// int deq[NMAX];
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]);
    }
    // int left_deq = 1, right_deq = 0;
 
    for(int i = 1; i <= n; i++) {
        // while(right_deq >= left_deq && v[i] <= v[deq[right_deq]]) {
        //     right_deq--;
        // }

        while (deq.empty() == false && v[i] <= v[deq.back()]) {
             deq.pop_back();
        }


        // deq[++right_deq] = i;

        deq.push_back(i);


        if(i >= k) {
            // answer += v[deq[left_deq]];
            // if(deq[left_deq] == i - k + 1) {
            //     left_deq++;
            // }

            answer += v[deq.front()];
            if(deq.front() == i - k + 1) {
                deq.pop_front();
            }
        }
    }
 
    printf("%lld\n", answer);
 
    return 0;
}