Cod sursa(job #2708945)

Utilizator pielevladutPiele Vladut Stefan pielevladut Data 19 februarie 2021 16:35:15
Problema Deque Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.62 kb
#include <bits/stdc++.h>

using namespace std;

int n, k;
int dq[5000001];
long long v[5000001];
long long sum = 0;

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

int main()
{
    fin >> n >> k;
    for(int i = 1; i <= n; i ++)
    {
        fin >> v[i];
    }
    int st = 0, dr = -1;
    for(int i = 1; i <= n; i ++)
    {
        if(st <= dr && dq[st] == i - k)
        {
            st++;
        }

        while(st <= dr && v[dq[dr]] >= v[i])
            dr--;

        dq[++dr] = i;

        if(st <= dr && i >= k)
            sum = sum + v[dq[st]];
    }
    fout << sum << '\n';
}