4333 2023. 03. 25 11:03:05 thetuberbee Sípálya (55 pont) cpp17 Elfogadva 55/55 115ms 21212 KiB
#include <bits/stdc++.h>

using namespace std;

int main ()
{
    int n, k;
    cin >> n >> k;
    vector<long long> vec(n);
    for (int i = 0; i < n; i++) {
        cin >> vec[i];
        vec[i] += i;
    }
    multiset<long long> maxset;
    long long sum = 0;
    for (int i = 0; i < k; i++) {
        sum += vec[i];
        maxset.insert(vec[i]);
    }
    //for (auto x : maxset) cout << x << " ";
    //cout << "\n";
    long long mincost = INT64_MAX;
    for (int i = k - 1; i < n; i++) {
        if (i >= k) {
            sum -= vec[i - k];
            sum += vec[i];
            maxset.erase(maxset.find(vec[i - k]));
            maxset.insert(vec[i]);
        }
        auto it = maxset.end();
        it--;
        long long localmax = *it;
        long long cost = k * localmax - sum;
        if (cost < mincost) {
            mincost = cost;
        }
        /*cout << localmax << " " << sum << " " << cost << "\nset: ";
        for (long long x : maxset) {
            cout << x << " ";
        }
        cout << "\n";*/
    }
    cout << mincost;
}
Részfeladat Összpont Teszt Verdikt Idő Memória
base 55/55
1 Elfogadva 0/0 3ms 1688 KiB
2 Elfogadva 0/0 3ms 1844 KiB
3 Elfogadva 2/2 3ms 2100 KiB
4 Elfogadva 2/2 3ms 2312 KiB
5 Elfogadva 2/2 3ms 2528 KiB
6 Elfogadva 2/2 3ms 2664 KiB
7 Elfogadva 3/3 3ms 2632 KiB
8 Elfogadva 1/1 8ms 2964 KiB
9 Elfogadva 1/1 8ms 2992 KiB
10 Elfogadva 1/1 8ms 3380 KiB
11 Elfogadva 1/1 8ms 3428 KiB
12 Elfogadva 1/1 8ms 3596 KiB
13 Elfogadva 1/1 8ms 4040 KiB
14 Elfogadva 2/2 8ms 4168 KiB
15 Elfogadva 2/2 7ms 3964 KiB
16 Elfogadva 2/2 108ms 7660 KiB
17 Elfogadva 2/2 112ms 16456 KiB
18 Elfogadva 2/2 114ms 17448 KiB
19 Elfogadva 3/3 112ms 21212 KiB
20 Elfogadva 2/2 112ms 8864 KiB
21 Elfogadva 2/2 109ms 8812 KiB
22 Elfogadva 2/2 111ms 8800 KiB
23 Elfogadva 2/2 111ms 8276 KiB
24 Elfogadva 2/2 115ms 8408 KiB
25 Elfogadva 2/2 112ms 9292 KiB
26 Elfogadva 2/2 112ms 9716 KiB
27 Elfogadva 2/2 112ms 10016 KiB
28 Elfogadva 3/3 112ms 9972 KiB
29 Elfogadva 3/3 112ms 10488 KiB
30 Elfogadva 3/3 114ms 11008 KiB