Cod sursa(job #2253194)

Utilizator AnDrEeA1915Monea Andreea AnDrEeA1915 Data 3 octombrie 2018 19:06:33
Problema Heapuri Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 0.57 kb
#include <bits/c++io.h>
#include <fstream>

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

#define nmax 5000010

int n, k, v[nmax], deque[nmax], start, finish;
long long sum;

int main()
{
    int i;
    fin >> n >> k;
    for (i = 1; i <= n; ++i)
        fin >> v[i];
    start = 1, finish = 0;
    for (i = 1; i <= n; ++i)
    {
        while (start <= finish && v[i] <= v[deque[finish]]) finish--;
        deque[++finish] = i;
        if (deque[start] == i-k) start++;
        if (i >= k) sum += v[deque[start]];
    }
    fout << sum;
    return 0;
}