Cod sursa(job #1978391)

Utilizator MaligMamaliga cu smantana Malig Data 7 mai 2017 17:16:31
Problema Deque Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.9 kb
#include <iostream>
#include <fstream>
#include <deque>

using namespace std;
ifstream in("deque.in");
ofstream out("deque.out");

#define pb push_back
#define ll long long
const int NMax = 5e6 + 5;
const int inf = 1e9 + 5;

int N,K;
int v[NMax];

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

    for (int i=1;i <= N;++i) {
        in>>v[i];
    }

    deque<int> deq;
    for (int i=1;i <= K;++i) {
        while (deq.size() && v[deq.back()] >= v[i]) {
            deq.pop_back();
        }

        deq.push_back(i);
    }

    ll sum = v[deq.front()];
    for (int i=K+1;i <= N;++i) {
        if (deq.front() < i-K+1) {
            deq.pop_front();
        }

        while (deq.size() && v[deq.back()] >= v[i]) {
            deq.pop_back();
        }
        deq.push_back(i);

        sum += v[deq.front()];
    }

    out<<sum<<'\n';

    in.close();out.close();
    return 0;
}