5702 | 2023. 09. 09 17:40:22 | TomaSajt | Sípálya (55 pont) | cpp17 | Elfogadva 55/55 | 32ms | 12728 KiB |
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
vector<ll> range_maxes(const vector<ll>& v, int k) {
int n = v.size();
vector<ll> res(n - k + 1);
deque<int> dq;
for (int i = 0; i < n; i++) {
while (!dq.empty() && i - dq.front() >= k) dq.pop_front();
while (!dq.empty() && v[dq.back()] <= v[i]) dq.pop_back();
dq.push_back(i);
if (i - k + 1 >= 0) {
res[i - k + 1] = v[dq.front()];
}
}
return res;
}
int main() {
cin.tie(0), ios::sync_with_stdio(0);
int n, k;
cin >> n >> k;
vector<ll> v(n);
for (auto& a : v) cin >> a;
for (int i = 0; i < n; i++) v[i] += i; // add the index, so that we now need to make flat regions instead of stairs
vector<ll> maxes = range_maxes(v, k); // maxes[i]: max value in the i-th K length interval
vector<ll> ps(n + 1);
partial_sum(v.begin(), v.end(), ps.begin() + 1);
ll best = LLONG_MAX;
for (int i = 0; i <= n - k; i++) {
ll sum = ps[i + k] - ps[i]; // sum of values in the i-th K length interval
ll cost = k * maxes[i] - sum;
best = min(best, cost);
}
cout << best;
}
Részfeladat | Összpont | Teszt | Verdikt | Idő | Memória | ||
---|---|---|---|---|---|---|---|
base | 55/55 | ||||||
1 | Elfogadva | 0/0 | 3ms | 1824 KiB | |||
2 | Elfogadva | 0/0 | 3ms | 2056 KiB | |||
3 | Elfogadva | 2/2 | 3ms | 2280 KiB | |||
4 | Elfogadva | 2/2 | 3ms | 2508 KiB | |||
5 | Elfogadva | 2/2 | 3ms | 2464 KiB | |||
6 | Elfogadva | 2/2 | 3ms | 2692 KiB | |||
7 | Elfogadva | 3/3 | 3ms | 2908 KiB | |||
8 | Elfogadva | 1/1 | 4ms | 3464 KiB | |||
9 | Elfogadva | 1/1 | 4ms | 3572 KiB | |||
10 | Elfogadva | 1/1 | 4ms | 3668 KiB | |||
11 | Elfogadva | 1/1 | 4ms | 3792 KiB | |||
12 | Elfogadva | 1/1 | 4ms | 3788 KiB | |||
13 | Elfogadva | 1/1 | 4ms | 3652 KiB | |||
14 | Elfogadva | 2/2 | 4ms | 3636 KiB | |||
15 | Elfogadva | 2/2 | 4ms | 3668 KiB | |||
16 | Elfogadva | 2/2 | 30ms | 12700 KiB | |||
17 | Elfogadva | 2/2 | 29ms | 11200 KiB | |||
18 | Elfogadva | 2/2 | 29ms | 11200 KiB | |||
19 | Elfogadva | 3/3 | 29ms | 10452 KiB | |||
20 | Elfogadva | 2/2 | 30ms | 12264 KiB | |||
21 | Elfogadva | 2/2 | 30ms | 12660 KiB | |||
22 | Elfogadva | 2/2 | 30ms | 12620 KiB | |||
23 | Elfogadva | 2/2 | 32ms | 12728 KiB | |||
24 | Elfogadva | 2/2 | 32ms | 12728 KiB | |||
25 | Elfogadva | 2/2 | 30ms | 12332 KiB | |||
26 | Elfogadva | 2/2 | 29ms | 12332 KiB | |||
27 | Elfogadva | 2/2 | 30ms | 12336 KiB | |||
28 | Elfogadva | 3/3 | 32ms | 12360 KiB | |||
29 | Elfogadva | 3/3 | 30ms | 12352 KiB | |||
30 | Elfogadva | 3/3 | 30ms | 12564 KiB |