Pagini recente » Cod sursa (job #386707) | Cod sursa (job #210272) | Rating Ghinescu Stefan-George (Stefan_Ghinescu) | Cod sursa (job #2045465) | Cod sursa (job #3257370)
#include <fstream>
#include <vector>
#include <deque>
std::string file_name = "deque";
std::ifstream fin(file_name + ".in");
std::ofstream fout(file_name + ".out");
long long sum;
int A[5000000], n, k;
std::deque<int> dq;
int main() {
fin >> n >> k;
for (int i = 1; i <= n; ++i) {
fin >> A[i];
}
for (int i = 1; i <= n; ++i) {
// Verificam daca ultimul element este in intervalul [i, i + k]
if (!dq.empty() && dq.front() < i - k + 1) {
// Daca nu, il eliminam
dq.pop_front();
}
// Acum eliminam toate elementele mai mici ca v[i]
// Facem acest lucru pentru ca vrem sa aflam doar minimul
// Si astfel minimul va fi la pozitia A[dq.front()]
while (!dq.empty() && A[dq.back()] >= A[i]) {
dq.pop_back();
}
// Adaugam noul indice in deque
dq.push_back(i);
// Verificam daca am parcurs cele k elemente
if (i >= k) {
sum += A[dq.front()];
}
}
fout << sum;
return 0;
}