Cod sursa(job #2664067)

Utilizator Tudor_PascaTudor Pasca Tudor_Pasca Data 27 octombrie 2020 20:48:13
Problema Deque Scor 60
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.75 kb
#include <iostream>
#include <fstream>
#include <deque>

using namespace std;

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

int N, K;
int64_t ans;
deque<pair<int64_t, int>> dq;

int main()
{
    in >> N >> K;

    for(int i = 1; i <= N; i++)
    {
        int64_t x;
        in >> x;

        //cout << i << ' ' << x << ' ' << ans << '\n';

        while(!dq.empty() && dq.back().first >= x)
            dq.pop_back();

        dq.emplace_back(x, i);

        if(!dq.empty() && dq.front().second  <= i - K)
            dq.pop_front();

        if(i >= K)
        {
            ans += dq.front().first;
            //cout << i << ' ' << dq.front().first << '\n';
        }
    }

    out << ans << '\n';

    return 0;
}