40032023-03-08 08:32:431478Sípálya (55 pont)cpp17Elfogadva 55/55120ms24168 KiB
#include <bits/stdc++.h>

using namespace std;

int main()
{
    //ifstream cin("in.txt");
    long long n;
    cin >> n;
    long long k;
    cin >> k;
    vector<long long> heights(n + 2);
    vector<long long> ps(n + 1);
    for(int i = 1; i <= n; i++){
        cin >> heights[i];
        ps[i] = ps[i - 1] + heights[i];
    }

    multiset<long long> s;
    for(int i = 1; i <= k; i++){
        s.insert(heights[i] + i);
    }

    long long ans = LONG_MAX;
    for(long long i = k; i <= n; i++){
        long long actualsum = ps[i] - ps[i - k];
        long long neededsum = (*s.rbegin()- i) * k + (k - 1)*k/2;
        //cout << i << " " << heights[i] << " " << *s.rbegin() - i << " " << neededsum - actualsum << '\n';
        ans = min(ans, neededsum - actualsum);
        s.erase(s.lower_bound(heights[i - k + 1] + i - k + 1));
        s.insert(heights[i + 1] + i + 1);
    }
    cout << ans;



    return 0;
}
RészfeladatÖsszpontTesztVerdiktIdőMemória
base55/55
1Elfogadva0/03ms1812 KiB
2Elfogadva0/03ms2184 KiB
3Elfogadva2/23ms2292 KiB
4Elfogadva2/23ms2452 KiB
5Elfogadva2/24ms2676 KiB
6Elfogadva2/23ms2892 KiB
7Elfogadva3/33ms3096 KiB
8Elfogadva1/18ms3856 KiB
9Elfogadva1/18ms3892 KiB
10Elfogadva1/18ms3856 KiB
11Elfogadva1/18ms4212 KiB
12Elfogadva1/18ms3928 KiB
13Elfogadva1/18ms4156 KiB
14Elfogadva2/28ms4264 KiB
15Elfogadva2/28ms4400 KiB
16Elfogadva2/2109ms10576 KiB
17Elfogadva2/2114ms19440 KiB
18Elfogadva2/2114ms20480 KiB
19Elfogadva3/3116ms24168 KiB
20Elfogadva2/2115ms11816 KiB
21Elfogadva2/2111ms11828 KiB
22Elfogadva2/2112ms11968 KiB
23Elfogadva2/2114ms11628 KiB
24Elfogadva2/2120ms11844 KiB
25Elfogadva2/2115ms12596 KiB
26Elfogadva2/2114ms12904 KiB
27Elfogadva2/2112ms13392 KiB
28Elfogadva3/3114ms13396 KiB
29Elfogadva3/3118ms13960 KiB
30Elfogadva3/3115ms14496 KiB