Submission #574654

#	Time	Username	Problem	Language	Result	Execution time	Memory
574654		four_specks	Stove (JOI18_stove)	C++17	100 / 100	19 ms	2900 KiB

stove

#include <bits/stdc++.h>

using namespace std;

inline namespace
{
} // namespace

void solve()
{
    int n, k;
    cin >> n >> k;

    vector<long> t(n);
    for (long &x : t)
        cin >> x;

    vector<long> diff(n - 1);
    for (long i = 0; i < n - 1; i++)
        diff[i] = t[i + 1] - t[i] - 1;
    sort(diff.begin(), diff.end(), greater<>());

    long res = t[n - 1] + 1 - t[0];
    for (int i = 0; i < k - 1; i++)
        res -= diff[i];

    cout << res << '\n';
}

int main()
{
    ios_base::sync_with_stdio(false), cin.tie(NULL);

    solve();

    return 0;
}

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