Cod sursa(job #2277748)

Utilizator cezar.plescaCezar Plesca cezar.plesca Data 6 noiembrie 2018 19:42:00
Problema Deque Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.17 kb
#include<stdio.h>
#include<stdlib.h>

#include<vector>
#include<deque>
#include<algorithm>

using namespace std;

#define MAXN 5000000

//int V[MAXN];
int* V;

struct pair_compare {
    bool operator() (const std::pair<int,int> & p1, const std::pair<int,int> & p2) const {
		if(p1.first == p2.first)
			return (p1.second < p2.second);
		else
			return (p1.first < p2.first);
    }
}pair_compare;

deque<pair<int,int>> multime;

int N, K;
long long rez;

void findsum(){
	
	rez+=(long long)(multime.front().first);
	
	for(int i=K; i<N; i++){
		while(!multime.empty()){
			if(multime.back().first>=V[i])
				multime.pop_back();
			else
				break;
		}
		multime.push_back(make_pair(V[i],i));
		
		while(multime.front().second<=(i-K))
			multime.pop_front();
		
		rez+=(long long)(multime.front().first);
	}
}

int main(){
	
	freopen("deque.in","rt",stdin);
	freopen("deque.out","wt",stdout);
	
	scanf("%d %d",&N,&K);
	V=new int[N];

	for(int i=0;i<N;i++){
		scanf("%d",&V[i]);
		if(i<K)
			multime.push_back(make_pair(V[i],i));
	}
	
	sort(multime.begin(),multime.end()); //,pair_compare);
	findsum();

	printf("%lld",rez);

	return 0;
}