Cod sursa(job #3205927)

Utilizator TeddyDinutaDinuta Eduard Stefan TeddyDinuta Data 21 februarie 2024 00:54:15
Problema Deque Scor 25
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.68 kb
#include <bits/stdc++.h>

using namespace std;
ifstream in("deque.in");
ofstream out("deque.out");
int n, k, a[5000005];
int main()
{
    ios_base::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    in >> n >> k;
    for (int i = 1; i <= n; i++) {
        in >> a[i];
    }

    deque<pair<int, int>> q;
    int sum = 0;

    for (int i = 1; i <= n; i++) {
        while(!q.empty() && q.back().first >= a[i])
            q.pop_back();
        while (!q.empty() && i - q.front().second >= k)
            q.pop_front();
        q.push_back({a[i], i});

        if (i >= k) {
            sum += q.front().first;
        }
    }

    out << sum << '\n';
    return 0;
}