Cod sursa(job #3170660)

Utilizator juniorOvidiu Rosca junior Data 17 noiembrie 2023 22:54:16
Problema Deque Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.04 kb
#include <fstream>
#include <deque>
#include <cstdint>

const int maxn = 5000010;

using namespace std;

int N, K;
int A[maxn];
deque<int> dq;
int64_t Sum = 0;

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

    fin >> N >> K;

    // Citesc șirul dat
    for (int i = 1; i <= N; i++) {
        fin >> A[i];
    }

    for (int i = 1; i <= N; i++) {
        // Cât timp elementul curent este mai mic decât ultimul element din deque, eliminăm pozitia ultimului element din deque
        while (not dq.empty() and A[i] <= A[dq.back()]) {
            dq.pop_back();
        }
        // Adaugăm poziția elementului curent în deque
        dq.push_back(i);

        // Dacă elementul minim coincide cu cel de pe poziția i-K, îi eliminăm poziția din deque
        if (dq.front() == i - K) {
            dq.pop_front();
        }

        // Adăugăm minimul la soluție
        if (i >= K) {
            Sum += A[dq.front()];
        }
    }

    fout << Sum << '\n';

    return 0;
}