Cod sursa(job #2842296)

Utilizator CiuiGinjoveanu Dragos Ciui Data 31 ianuarie 2022 15:29:20
Problema Deque Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.06 kb
#include <iostream>
#include <fstream>
#include <cmath>
#include <deque>
using namespace std;
 
ifstream fin("deque.in");
ofstream fout("deque.out");
 
const int MAX_SIZE = 5000005;
int v[MAX_SIZE];
 
int main() {
    int n, k;
    fin >> n >> k;
    for (int i = 1; i <= n; ++i) {
        fin >> v[i];
    }
    deque<int> minimums, minimumsPositions;
    long long sum = 0;
    for (int i = 1; i <= n + 1; ++i) {
        while (!minimums.empty() && minimums.back() > v[i]) {
            minimums.pop_back();
            minimumsPositions.pop_back();
        }
        if (i > k) {
            //cout << numbers.front() << " ";
            sum += minimums.front();
        }
        if (!minimums.empty() && i - k >= minimumsPositions.front()) {
            minimumsPositions.pop_front();
            minimums.pop_front();
        }
        minimums.push_back(v[i]);
        minimumsPositions.push_back(i);
    
    }
    fout << sum;
}
 
/*
 
 1 1
 -1 -> -1
 
 5 1
 1 2 3 4 5 -> 15
 
 9 3
 -7 9 2 4 -1 5 6 7 1 -> -2
 
 9 3
 -7 9 2 4 -1 5 6 7 10 -> 3
 
 4 4
 1 -1 2 3 -> -1
 
 
 
 */