Cod sursa(job #796275)

Utilizator toranagahVlad Badelita toranagah Data 10 octombrie 2012 22:12:20
Problema Deque Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 0.99 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;
long long 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];
    }
    int result = 0;
    for (int i = 1; i <= N; ++i) {
        if (!dq.empty() && i > K) {
            // fout << endl << sir[dq.front()] << endl;
            result += sir[dq.front()];
        }
        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);
        // for (it = dq.begin(), eit = dq.end(); it != eit; ++it) {
        //     fout << sir[*it] << ' ';
        // }
        // fout << endl;
    }
    while (!dq.empty()) {
        result += sir[dq.front()];
        dq.pop_back();
    }
    fout << result;
    return 0;
}