Cod sursa(job #3256644)

Utilizator nicolasschlachter740@gmail.comNicolas Schlachter [email protected] Data 15 noiembrie 2024 17:19:32
Problema Deque Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.68 kb
#include <bits/stdc++.h>
using namespace std;

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

    int N, K;
    fin >> N >> K;
    vector<int> A(N);
    for (int i = 0; i < N; ++i) {
        fin >> A[i];
    }

    deque<int> dq; 
    long long suma = 0;

    for (int i = 0; i < N; ++i) {
        if (!dq.empty() && dq.front() < i - K + 1) {
            dq.pop_front();
        }

        while (!dq.empty() && A[dq.back()] >= A[i]) {
            dq.pop_back();
        }

        
        dq.push_back(i);

        if (i >= K - 1) {
            suma += A[dq.front()];
        }
    }

    fout << suma << "\n";

    return 0;
}