Cod sursa(job #1677469)

Utilizator sandupetrascoPetrasco Sandu sandupetrasco Data 6 aprilie 2016 16:44:09
Problema Deque Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.68 kb
#include <bits/stdc++.h>

using namespace std;
int A[5000002];
int N, K;
long long sum;
deque <int> D;
void add(int x){
    while(!D.empty() && A[D.back()] >= A[x])
        D.pop_back(); 
    D.push_back(x);
    while(x - D.front() + 1 > K) D.pop_front();
}
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);
    assert(freopen("deque.in", "r", stdin));
    assert(freopen("deque.out", "w", stdout));
    cin >> N >> K;
    for(int i = 1; i <= N; i++) cin >> A[i];
    for(int i = 1; i <= K; i++){
        add(i);}
        sum += A[D.front()];
    for(int i = K+1; i <= N; i++){
        add(i);
        sum += A[D.front()];
    }
    cout << sum;
    return 0;
}