Submission #587577

#	Time	Username	Problem	Language	Result	Execution time	Memory
587577		dryeab	Stove (JOI18_stove)	C++17	100 / 100	30 ms	2056 KiB

stove

#include <bits/stdc++.h>

using namespace std;

const int M = 1e5 + 5;

int t[M];
int gap[M];

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

    int n, k;
    cin >> n >> k;

    for (int i = 0; i < n; ++i)
        cin >> t[i];

    int res = n;

    if (k >= n)
    {
        cout << res << endl;
        return 0;
    }

    sort(t, t + n);
    for (int i = 1; i < n; ++i)
        gap[i - 1] = -(t[i] - t[i - 1] - 1);

    sort(gap, gap + n - 1);

    k--;
    for (int i = k; i < n - 1; ++i)
        res += -gap[i];

    cout << res << endl;
}

• Subtask 1: 20 / 20
• Subtask 2: 30 / 30
• Subtask 3: 50 / 50