219132026-01-14 10:19:53hunzombiSípálya (55 pont)python3Runtime error 0/5516ms3220 KiB
#include <bits/stdc++.h>
using namespace std;

typedef long long ll;

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);

    ll n, k;
    cin >> n >> k;

    vector<ll> vec(n);

    for (ll i=0; i < n; i++) cin >> vec[i];

    multiset<ll> ms;
    ll curr = 0;
    ll ans = LLONG_MAX;

    for (int i = 0; i < n; i++) {
        ms.insert(vec[i] + i);
        curr += vec[i];

        if (i >= k) {
            ms.erase(ms.find(vec[i - k] + (i - k)));
            curr -= vec[i - k];
        }

        if (i >= k - 1) {
            ll L = i - k + 1;
            ll H = *ms.begin() - L;

            ll reqHeight = k * H - k * (k - 1) / 2;
            ll cost = reqHeight - curr;
            if (cost >= 0)
                ans = min(ans, cost);
        }
    }

    cout << ans << '\n';

    return 0;
}
SubtaskSumTestVerdictTimeMemory
base0/55
1Runtime error0/016ms3076 KiB
2Runtime error0/016ms2860 KiB
3Runtime error0/216ms2868 KiB
4Runtime error0/216ms2868 KiB
5Runtime error0/216ms2892 KiB
6Runtime error0/216ms2936 KiB
7Runtime error0/316ms3052 KiB
8Runtime error0/116ms3036 KiB
9Runtime error0/116ms3220 KiB
10Runtime error0/116ms2868 KiB
11Runtime error0/116ms2868 KiB
12Runtime error0/116ms2868 KiB
13Runtime error0/114ms2852 KiB
14Runtime error0/216ms2900 KiB
15Runtime error0/214ms2980 KiB
16Runtime error0/216ms2948 KiB
17Runtime error0/216ms3040 KiB
18Runtime error0/214ms2868 KiB
19Runtime error0/316ms2976 KiB
20Runtime error0/214ms3044 KiB
21Runtime error0/216ms2868 KiB
22Runtime error0/216ms2908 KiB
23Runtime error0/214ms3020 KiB
24Runtime error0/214ms2868 KiB
25Runtime error0/216ms3036 KiB
26Runtime error0/214ms2868 KiB
27Runtime error0/214ms2888 KiB
28Runtime error0/316ms2868 KiB
29Runtime error0/314ms3060 KiB
30Runtime error0/316ms2824 KiB