Cod sursa(job #3002530)

Utilizator NeganAlex Mihalcea Negan Data 14 martie 2023 20:59:39
Problema Deque Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.79 kb
#include <bits/stdc++.h>

using namespace std;
ifstream fin("deque.in");
ofstream fout("deque.out");
int n, k;
int a[5000005];
deque<int>dq;
long long sol;


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