Cod sursa(job #3037785)

Utilizator divadddDavid Curca divaddd Data 26 martie 2023 13:54:40
Problema Deque Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.61 kb
#include <bits/stdc++.h>
using namespace std;
const int NMAX = 5e6+2;
deque<int> dq;
int n,a[NMAX],k;
long long ans;

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

signed main()
{
    fin >> n >> k;
    for(int i = 1; i <= n; i++){
        fin >> a[i];
    }
    for(int i = 1; i <= n; i++){
        while(!dq.empty() && a[dq.back()] >= a[i]){
            dq.pop_back();
        }
        dq.push_back(i);
        while(i-dq.front()+1 > k){
            dq.pop_front();
        }
        if(i >= k){
            ans += a[dq.front()];
        }
    }
    fout << ans;
    return 0;
}