Cod sursa(job #2760741)

Utilizator DariusMDarius Feher DariusM Data 28 iunie 2021 21:12:57
Problema Deque Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.57 kb
#include <iostream>
#include <fstream>
#include <deque>
using namespace std;

ifstream f("deque.in");
ofstream g("deque.out");

deque <long long> deq;

int arr[5000001];

int main() {
	int n, k;
	f >> n >> k;
	for (long long i = 1; i <= n; i++) {
		f >> arr[i];
	}
	long long sum = 0, back;
	for (int i = 1; i <= n; i++) {
		back = deq.size() - 1;
		while(back >= 0 && arr[i] <= arr[deq.at(back--)]) {
			deq.pop_back();
		}
		deq.push_back(i);
		if (deq.front() == i - k) {
			deq.pop_front();
		}
		if (i >= k) {
			sum += arr[deq.front()];
		}
	}
	g << sum << "\n";
	return 0;
}