Cod sursa(job #927839)

Utilizator VladMSBonta vlad valentin VladMS Data 26 martie 2013 08:31:27
Problema Deque Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.99 kb
#include <fstream>
#define maxn 5000010
using namespace std;
int n,k;
int a[maxn], deque[maxn];
int pr,u;
long long sum;

int main()
{
    ifstream fin("deque.in");
    ofstream fout("deque.out");
    int i;
    fin>>n>>k;
    // Citesc sirul dat
    for (i = 1; i <= n; ++i)
        fin>>a[i];
    u=0;
    pr=1;
    for (i = 1; i <= n; i++)
    {
        // Cat timp elementul curent este mai mic decat ultimul element din deque, eliminam pozitia ultimului element din deque
        while (pr <= u && a[i] <= a[ deque[u] ])
            u--;
        // Adaugam pozitia elementului curent in deque
        deque[++u] = i;
        // Daca elementul minim coincide cu cel de pe pozita i-K, ii eliminam pozitia din deque, deoarece acesta nu mai conteaza pentru pasii >= i
        if (deque[pr] == i-k)
            pr++;
        // Afisam minimul, acesta aflandu-se in varful deque-ului
        if (i >= k)
            sum +=a[ deque[pr]];
    }
    fout<<sum;
    return 0;
}