4333 | 2023. 03. 25 11:03:05 | thetuberbee | Sípálya (55 pont) | cpp17 | Elfogadva 55/55 | 115ms | 21212 KiB |
#include <bits/stdc++.h>
using namespace std;
int main ()
{
int n, k;
cin >> n >> k;
vector<long long> vec(n);
for (int i = 0; i < n; i++) {
cin >> vec[i];
vec[i] += i;
}
multiset<long long> maxset;
long long sum = 0;
for (int i = 0; i < k; i++) {
sum += vec[i];
maxset.insert(vec[i]);
}
//for (auto x : maxset) cout << x << " ";
//cout << "\n";
long long mincost = INT64_MAX;
for (int i = k - 1; i < n; i++) {
if (i >= k) {
sum -= vec[i - k];
sum += vec[i];
maxset.erase(maxset.find(vec[i - k]));
maxset.insert(vec[i]);
}
auto it = maxset.end();
it--;
long long localmax = *it;
long long cost = k * localmax - sum;
if (cost < mincost) {
mincost = cost;
}
/*cout << localmax << " " << sum << " " << cost << "\nset: ";
for (long long x : maxset) {
cout << x << " ";
}
cout << "\n";*/
}
cout << mincost;
}
Részfeladat | Összpont | Teszt | Verdikt | Idő | Memória | ||
---|---|---|---|---|---|---|---|
base | 55/55 | ||||||
1 | Elfogadva | 0/0 | 3ms | 1688 KiB | |||
2 | Elfogadva | 0/0 | 3ms | 1844 KiB | |||
3 | Elfogadva | 2/2 | 3ms | 2100 KiB | |||
4 | Elfogadva | 2/2 | 3ms | 2312 KiB | |||
5 | Elfogadva | 2/2 | 3ms | 2528 KiB | |||
6 | Elfogadva | 2/2 | 3ms | 2664 KiB | |||
7 | Elfogadva | 3/3 | 3ms | 2632 KiB | |||
8 | Elfogadva | 1/1 | 8ms | 2964 KiB | |||
9 | Elfogadva | 1/1 | 8ms | 2992 KiB | |||
10 | Elfogadva | 1/1 | 8ms | 3380 KiB | |||
11 | Elfogadva | 1/1 | 8ms | 3428 KiB | |||
12 | Elfogadva | 1/1 | 8ms | 3596 KiB | |||
13 | Elfogadva | 1/1 | 8ms | 4040 KiB | |||
14 | Elfogadva | 2/2 | 8ms | 4168 KiB | |||
15 | Elfogadva | 2/2 | 7ms | 3964 KiB | |||
16 | Elfogadva | 2/2 | 108ms | 7660 KiB | |||
17 | Elfogadva | 2/2 | 112ms | 16456 KiB | |||
18 | Elfogadva | 2/2 | 114ms | 17448 KiB | |||
19 | Elfogadva | 3/3 | 112ms | 21212 KiB | |||
20 | Elfogadva | 2/2 | 112ms | 8864 KiB | |||
21 | Elfogadva | 2/2 | 109ms | 8812 KiB | |||
22 | Elfogadva | 2/2 | 111ms | 8800 KiB | |||
23 | Elfogadva | 2/2 | 111ms | 8276 KiB | |||
24 | Elfogadva | 2/2 | 115ms | 8408 KiB | |||
25 | Elfogadva | 2/2 | 112ms | 9292 KiB | |||
26 | Elfogadva | 2/2 | 112ms | 9716 KiB | |||
27 | Elfogadva | 2/2 | 112ms | 10016 KiB | |||
28 | Elfogadva | 3/3 | 112ms | 9972 KiB | |||
29 | Elfogadva | 3/3 | 112ms | 10488 KiB | |||
30 | Elfogadva | 3/3 | 114ms | 11008 KiB |