Cod sursa(job #2888817)

Utilizator TindecheTindeche Alexandru Tindeche Data 11 aprilie 2022 21:00:48
Problema Deque Scor 25
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.87 kb
#include <string.h>
#include <deque>
#include <fstream>
#include <iostream>

#define n_max 5000004

using namespace std;

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

/*
 * Simulam un deque folosind un array. dqom pastra intotdeauna pe prima pozitie minimul local si scoatem pe rand ce este
 * pe prima pozitie si este prea departe
 */

int v[n_max];
int dq[n_max];

int main ()
{
    int n, k, sum = 0;
    fin >> n >> k;
    int l = 1, r = 0;
    for(int i = 1; i <= n; i++)
        fin >> v[i];
    for(int i = 1; i <= n; i++)
    {
        while(r >= l && v[dq[r]] >= v[i])      // Cand r < l inseamna ca stidqa e goala
        {
            r--;
        }
        dq[++r] = i;
        if(i >= k)
        {
            while(r >= l && dq[l] <= i - k)
            {
                l++;
            }
            sum += v[dq[l]];
        }
    }
    fout << sum;
    return 0;
}