Cod sursa(job #468329)

Utilizator nashnash mit nash Data 3 iulie 2010 10:11:26
Problema Deque Scor 60
Compilator cpp Status done
Runda Arhiva educationala Marime 1.14 kb
#include <stdlib.h>
#include <cstdio>
#include <algorithm>
#include <cstring>

using namespace std;

int aib[5000001];
int vec[5000000];
int n,m,nr,o,a,b;

#define ZERO(x) ( (x) & (-x) )
#define INF (0x3f3f3f3f)

inline int query(int a,int b) {
    int vmin = INF;
    if(a>b) return vmin;

    for(int i = b ; i >= a ; )
        if(i - ZERO(i) + 1 >= a )
            vmin = min( vmin , aib[i]) , i -= ZERO(i);
        else
            vmin = min( vmin , vec[i]) , i--;

    return vmin;
}

inline void construct(int p, int nr) {

    vec[p] = nr;

    aib[p] = min( aib[p] , min( query(p - ZERO(p) + 1 , p - 1 ) , vec[p] ) );

    if( p + ZERO(p) <= n )
        aib[ p + ZERO(p) ] = min( aib[p + ZERO(p) ] , vec[p] );

}

int main(int argc, char** argv) {

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

    scanf("%d %d",&n,&m);

    for(int i = 0 ; i <= n ; i++ )
        aib[ i ] = INF;

    for(int i = 1 ; i <= n ; i++) {
        scanf("%d",&nr);
        construct(i,nr);
    }

    long long sum = 0;
    for(int i = 1 ; i + m - 1 <= n ; i++ ) {
        sum += query( i , i + m - 1 );
    }

    printf("%lld\n",sum);

    return 0;
}