Cod sursa(job #1350219)

Utilizator IonSebastianIon Sebastian IonSebastian Data 20 februarie 2015 18:33:36
Problema Deque Scor 25
Compilator cpp Status done
Runda Arhiva educationala Marime 1.05 kb
#include <cstdio>
//#include <deque>

using namespace std;

const int MAX_N = 5000000;

FILE *in, *out;

//deque<int> dq;

int dq[MAX_N], st, dr;

int v[MAX_N];

int main()
{
    in = fopen("deque.in","r");
    out = fopen("deque.out", "w");
    int n, k;
    int i;
    int sum = 0;
    fscanf(in, "%d%d", &n, &k);
    for(i = 0; i < n; i++)
        fscanf(in, "%d", &v[i]);
    dr = 0;
    st = 1;
    for(i = 0; i < n; i++)
    {
        while(st <= dr && v[i] < v[dq[dr]]) dr--;
        dq[++dr] = i;
        if(dq[st] == i-k) st++;
        if(i >= k-1) sum += v[dq[st]];
    }
    /*i = 0;
    while(i < k)
    {
        while((!dq.empty()) && (v[dq.back()] > v[i])) dq.pop_back();
        dq.push_back(i);
        i++;
    }
    sum = v[dq.front()];
    for(int i = k; i < n; i++)
    {
        if((!dq.empty()) && (dq.front() <= i-k)) dq.pop_front();
        while((!dq.empty()) && (v[dq.back()] > v[i])) dq.pop_back();
        dq.push_back(i);
        sum += v[dq.front()];
    }*/
    fprintf(out, "%d", sum);
    fclose(in);
    fclose(out);
    return 0;
}