Cod sursa(job #1245576)

Utilizator EpictetStamatin Cristian Epictet Data 19 octombrie 2014 15:24:33
Problema Deque Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.92 kb
#include <fstream>
#include <deque>
#define Nmax 5000010
using namespace std;
ifstream fin ("deque.in");
ofstream fout ("deque.out");
int N, K, A[Nmax];
long long sum;
deque < int > Dq;

int main()
{
    fin >> N >> K;
    for (int i=1; i<=N; i++) fin >> A[i];

    for (int i=1; i<=N; i++)
    {
        // Cat timp elementul curent este mai mic decat ultimul element din deque, eliminam pozitia ultimului element din deque;
        while (!Dq.empty() && A[i] <= A[Dq.back()])
            Dq.pop_back();
        // Adaugam pozitia elementului curent in deque;
        Dq.push_back(i);
        // Daca elementul nu mai apartine secventei curente (pozitia lui este mai mica sau egala cu i-K), ii eliminam pozitia din deque;
        while (Dq.front() <= i-K) Dq.pop_front();
        // Adunam minimul din varful deque-ului la solutie;
        if (i >= K) sum += A[Dq.front()];
    }

    fout << sum << '\n';
    fout.close();
    return 0;
}