Cod sursa(job #2269924)

Utilizator st_marianStoica Marian st_marian Data 26 octombrie 2018 19:41:07
Problema Deque Scor 60
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.96 kb
#include <bits/stdc++.h>
#define NMAX 5000001
using namespace std;
ifstream fin("deque.in");
ofstream fout("deque.out");
int n, k;
long long sum, vec[NMAX];
struct elem
{
    int val, poz;
};
deque<elem> mydeck;
int main()
{
    fin>>n>>k;
    for(int i=1; i<=n; i++) fin>>vec[i];
    for(int i=1; i<k; i++)
    {
        for(int j=mydeck.size()-1; j>=0; j--)
            if(vec[i]<=mydeck[j].val)    mydeck.pop_back();
            else break;
        mydeck.push_back(elem{});
        mydeck.back().poz=i;
        mydeck.back().val=vec[i];
    }
    for(int i=k; i<=n; i++)
    {
        for(int j=mydeck.size()-1; j>=0; j--)
            if(vec[i]<=mydeck[j].val)    mydeck.pop_back();
            else break;
        mydeck.push_back(elem{});
        mydeck.back().poz=i;
        mydeck.back().val=vec[i];
        if(mydeck.front().poz==i-k)    mydeck.pop_front();
        sum+=mydeck.front().val;
    }
    fout<<sum<<'\n';
    return 0;
}