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