Cod sursa(job #1978406)

Utilizator BogdanisarBurcea Bogdan Madalin Bogdanisar Data 7 mai 2017 17:36:16
Problema Deque Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.29 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];
// v - vectorul initial

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

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

    // minimul pentru o subsecventa de lungime K se va afla pe deq.front();
    // deque-ul va contine un subsir crescator al subsecventei de lungime K,
    // unde intre doua elementele ale subsirului nu se afla un altul mai mic sau egal cu ele,
    // primul element este minimul subsecventei,
    // ultimul element al subsirului este ultimul element al subsecventei
    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;
}