Cod sursa(job #2059507)

Utilizator efinagAdina Grecu efinag Data 7 noiembrie 2017 09:36:24
Problema Deque Scor 25
Compilator cpp Status done
Runda Arhiva educationala Marime 0.89 kb
#include <iostream>
#include <fstream>

using namespace std;
const int N=5000000;
int v[N];
int d[N];
int main()
{
    ifstream in("deque.in");
    ofstream out("deque.out");
    int st=0,n,k;
    int dr=-1;
    in>>n>>k;
    for (int i=0;i<n;i++)
    {
        in>>v[i];
    }
    int sum=0;
    for(int i=0; i<n; i++)
    {
        //elimin din stanga daca cel mai vechi nu mai apartine secventei curente
        if(st<=dr && d[st]==i-k)
        {
            st++;
        }

        //elimin din dreapta elemente mai vechi si mai proaste, care nu vor mai fi selectate pe viitor
        while (st<=dr && v[i]<=v[d[dr]])
        {
            dr--;
        }

        //adaug in deque elem curent
        d[++dr]=i;

        //actualizez rezulatul global
        if(i>=k-1) //am o secventa de lungime k care se term pe poz i
        {
            sum+=v[d[st]];
        }
    }
    out<<sum;

    return 0;
}