Cod sursa(job #2842292)

Utilizator CiuiGinjoveanu Dragos Ciui Data 31 ianuarie 2022 15:24:32
Problema Deque Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.04 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], n, k;
 
int main() {
    fin >> n >> k;
    for (int i = 1; i <= n; ++i) {
        fin >> v[i];
    }
    deque<int> numbers;
    deque<int> positions;
    long long sum = 0;
    for (int i = 1; i <= n; ++i) {
        while (!numbers.empty() && numbers.back() > v[i]) {
            numbers.pop_back();
            positions.pop_back();
        }
        if (i > k) {
            //cout << numbers.front() << " ";
            sum += numbers.front();
        }
        if (!numbers.empty() && i - k >= positions.front()) {
            positions.pop_front();
            numbers.pop_front();
        }
        numbers.push_back(v[i]);
        positions.push_back(i);
    
    }
    sum += numbers.front();
    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
 
 
 
 */