Cod sursa(job #2267123)

Utilizator andreiomd1Onut Andrei andreiomd1 Data 23 octombrie 2018 12:05:28
Problema Deque Scor 25
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.68 kb
#include <bits/stdc++.h>

using namespace std;

const int NMAX=(5*(1e6));

int N, K, i;

int A[NMAX+5];

deque <int> D;

long long ans;

int main()
{
    freopen("deque.in", "r", stdin);
    freopen("deque.out", "w", stdout);

    scanf("%d%d", &N, &K);

    for(i=1; i<=N; i++)
        scanf("%d", &A[i]);

    D.push_back(1);

    for(i=2; i<=N; i++)
    {
        //Analiza amortizata:
        while(!D.empty() && A[D.back()] > A[i])
            D.pop_back();

        D.push_back(i);

        if(D.back() - D.front() >= K)
            D.pop_front();

        if(i>=K)
            ans+=A[D.front()];
    }

    printf("%d\n", ans);

    return 0;
}