Cod sursa(job #2651975)

Utilizator AndreiAlexandru2k3Ciucan Andrei Alexandru AndreiAlexandru2k3 Data 23 septembrie 2020 21:49:29
Problema Deque Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.62 kb
#include <cstdio>
using namespace std;

FILE *f=fopen("deque.in","r");
FILE *g=fopen("deque.out","w");

const int NMAX=5000001;

int N,K,x;
int v[NMAX],DQ[NMAX];

long long solve()
{
    fscanf(f,"%d%d",&N,&K);
    for(int i=1;i<=N;i++)
        fscanf(f,"%d",&v[i]);
    long long S=0;
    int f=1,b=0;
    for(int i=1;i<=N;i++)
    {
        //
        while(f<=b&&v[i]<=v[DQ[b]])
            b--;
        //
        DQ[++b]=i;
        if(DQ[f]==i-K)
            f++;
        if(i>=K)
            S+=v[DQ[f]];
    }
    return S;
}

int main()
{
    fprintf(g,"%lld",solve());
    return 0;
}