8216 | 2024. 01. 12 20:35:36 | gergomiszori | Sípálya (55 pont) | cpp17 | Elfogadva 55/55 | 68ms | 21100 KiB |
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define endl "\n"
#define pll pair<ll,ll>
#define fs first
#define sc second
const ll MOD = 1e9+7;
void solve()
{
ll n, k;
cin >> n >> k;
vector<ll> a(n);
for(ll i = 0; i < n; i++)
{
cin >> a[i];
a[i] += i;
}
if(k == 1)
{
cout << 0 << endl;
return;
}
ll ans = LLONG_MAX;
multiset<ll> elements;
ll vsum = 0;
for(ll i = 0; i < k; i++)
{
vsum += a[i];
elements.insert(a[i]);
}
for(ll i = k-1; i < n; i++)
{
if(i >= k)
{
vsum -= a[i-k];
vsum += a[i];
elements.erase(elements.find(a[i-k]));
elements.insert(a[i]);
}
auto it = elements.end();
it--;
ans = min(ans, k * *it - vsum);
}
cout << ans << endl;
}
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(NULL); cout.tie(NULL);
ll t = 1;
//cin >> t;
while(t--)
{
solve();
}
return 0;
}
Részfeladat | Összpont | Teszt | Verdikt | Idő | Memória | ||
---|---|---|---|---|---|---|---|
base | 55/55 | ||||||
1 | Elfogadva | 0/0 | 3ms | 1824 KiB | |||
2 | Elfogadva | 0/0 | 3ms | 2064 KiB | |||
3 | Elfogadva | 2/2 | 3ms | 2416 KiB | |||
4 | Elfogadva | 2/2 | 3ms | 2580 KiB | |||
5 | Elfogadva | 2/2 | 3ms | 2612 KiB | |||
6 | Elfogadva | 2/2 | 3ms | 2748 KiB | |||
7 | Elfogadva | 3/3 | 3ms | 2832 KiB | |||
8 | Elfogadva | 1/1 | 6ms | 3180 KiB | |||
9 | Elfogadva | 1/1 | 6ms | 3388 KiB | |||
10 | Elfogadva | 1/1 | 6ms | 3276 KiB | |||
11 | Elfogadva | 1/1 | 6ms | 3432 KiB | |||
12 | Elfogadva | 1/1 | 6ms | 3420 KiB | |||
13 | Elfogadva | 1/1 | 6ms | 3772 KiB | |||
14 | Elfogadva | 2/2 | 6ms | 4084 KiB | |||
15 | Elfogadva | 2/2 | 6ms | 3852 KiB | |||
16 | Elfogadva | 2/2 | 63ms | 7396 KiB | |||
17 | Elfogadva | 2/2 | 68ms | 16036 KiB | |||
18 | Elfogadva | 2/2 | 65ms | 17172 KiB | |||
19 | Elfogadva | 3/3 | 68ms | 21100 KiB | |||
20 | Elfogadva | 2/2 | 65ms | 8900 KiB | |||
21 | Elfogadva | 2/2 | 64ms | 8352 KiB | |||
22 | Elfogadva | 2/2 | 64ms | 8356 KiB | |||
23 | Elfogadva | 2/2 | 64ms | 7844 KiB | |||
24 | Elfogadva | 2/2 | 64ms | 8100 KiB | |||
25 | Elfogadva | 2/2 | 67ms | 9112 KiB | |||
26 | Elfogadva | 2/2 | 67ms | 9596 KiB | |||
27 | Elfogadva | 2/2 | 64ms | 10012 KiB | |||
28 | Elfogadva | 3/3 | 65ms | 10072 KiB | |||
29 | Elfogadva | 3/3 | 65ms | 10328 KiB | |||
30 | Elfogadva | 3/3 | 68ms | 10868 KiB |