Cod sursa(job #796303)

Utilizator toranagahVlad Badelita toranagah Data 10 octombrie 2012 23:31:59
Problema Deque Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.69 kb
#include <fstream>
#include <deque>
using namespace std;

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

deque<int> dq;
deque<int>::const_iterator it, eit;
int sir[5000010];
int N, K;

int main(int argc, char const *argv[])
{
    fin >> N >> K;
    for (int i = 1; i <= N; ++i) {
        fin >> sir[i];
    }
    long long result = 0;
    for (int i = 1; i <= N; ++i) {
        while (!dq.empty() && dq.front() < i - K + 1) {
            dq.pop_front();
        }
        while (!dq.empty() && sir[i] < sir[dq.back()]) {
            dq.pop_back();
        }
        dq.push_back(i);
        if (i >= K) {
            result += sir[dq.front()];
        }
    }
    fout << result;
    return 0;
}